公开(公告)号：CN118735942A

公开(公告)日：2024-10-01

申请号：CN202410767592.3

申请日：2024-06-14

Applicant: 西安电子科技大学 ,  西安电子科技大学杭州研究院

Inventor： 高飞 ,  施斯 ,  蒋依凡 ,  何立火 ,  王楠楠

IPC: G06T7/11 ,  G06N3/0464 ,  G06N3/0455 ,  G06N3/048 ,  G06N3/084 ,  G06V10/26 ,  G06V10/764 ,  G06V10/774

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