一种基于MobileViT与UNet模型的图像分割方法及系统

    公开(公告)号:CN117726820A

    公开(公告)日:2024-03-19

    申请号:CN202311834754.2

    申请日:2023-12-28

    Applicant: 华侨大学

    Abstract: 本发明公开了一种基于MobileViT与UNet模型的图像分割方法及系统,应用图像识别领域,方法包括:将原始的RGB三通道图像大小进行调整,并对应调整标签图像大小;将调整后的RGB图像输入分割模型进行训练,获得训练好的模型参数;所述分割模型包括编码器和解码器;所述编码器用于对调整后的RGB图像进行特征提取;所述解码器的输入为所述编码器的输出,所述解码器输出与编码器部分相同空间分辨率的特征图;使用训练好的分割模型进行图像分割,输出与标签图像的大小一致的分割图像。本发明降低了模型的计算成本,提高了分割速度和分割精度,适用于实时和大规模应用。

Patent Agency Ranking