基于广义类平衡损失的图像分割方法

本发明公开了基于广义类平衡损失的图像分割方法，包括以下步骤；步骤1：将图像输入到分割网络中；通过分割网络中的编码器提取图像特征，并通过解码器将所述图像特征恢复到原始分辨率，生成预测的分割图像；步骤2：将真实的分割图像作为真实标签，计算所述预测的分割图像和真实的分割图像之间的广义类平衡损失；步骤3：利用所述广义类平衡损失进行反向传播，并更新网络参数。本发明利用不同的调节因子，有效平衡了不同类别间的损失贡献。通过精确地分割小样本区域，进一步提升了整体的图像分割精度，实现了在保证总体性能的同时，特别增强了模型对不均衡数据分布的处理能力。