一种基于SegFormer的海上图像语义分割方法

本发明公开了一种基于SegFormer的海上图像语义分割方法，S1：获取海上语义分割数据集；S2：基于SegFormer网络架构建立海上图像语义分割模型，基于所述海上语义分割数据集对海上图像语义分割模型进行训练，并设置混合损失函数作为训练过程中的损失函数，当损失函数收敛时，得到训练后的海上图像语义分割模型；S3：基于所述训练后的海上图像语义分割模型，对待识别的图像进行海上图像语义分割处理。本发明中的海上图像语义分割模型更加关注对物体边缘特征以及小尺寸目标特征提取的重要通道，避免类别严重不平衡的问题，最终提升了航海图像语义分割的稳定性和准确性，以满足不同环境和应用需求。