一种基于深度学习的轻量化图像语义分割方法

    公开(公告)号:CN116486080A

    公开(公告)日:2023-07-25

    申请号:CN202310376737.2

    申请日:2023-04-10

    Abstract: 本发明公开一种基于深度学习的轻量化图像语义分割方法,先构建轻量化图像语义分割模型Mobile‑SegFormer;再利用已分割好的样本图像集对构建的轻量化图像语义分割模型进行训练,得到训练好的轻量化图像语义分割模型;后将待分割的图像送入到训练好的轻量化图像语义分割模型中,训练好的轻量化图像语义分割模型输出分割好的图片。本发明将SegFormer和MobileNetv3模型都进行了改进,无需通过Imagenet数据集的预训练,即可实现较高的分割精度,具有较友好的可移植性。通过实验进行对比,我们提出的Mobile‑SegFormer网络,相对于常见的轻量化网络,有着更准确的分割效果,相对于大型的分割网络,有着更轻量化的参数量和更低的每秒浮点运算值,能够较好的均衡Params、FLOPS和MIoU指标。

Patent Agency Ranking