ISSN 1009-5624 CN 10-2021/TQ    主管:中国乐凯集团有限公司    主办:北京乐凯科技有限公司

中国知网全文收录期刊
万方数据库收录期刊
RCCSE中文学术期刊
维普资讯网/超星域出版 全文收录
中国核心期刊(遴选)数据库收录期刊
首页 > 刊期 > 2025 > 1期 > 综合:探索与发现
基于优化U⁃net 的单色图像分割算法
肖政文,刘 丹(通信作者)

【摘要】数字图像处理在公安领域的应用十分广泛,其中对单色图像分割预处理,在人脸识别、车辆识别和指纹识别等方面有着十分重要的作用。本文提出一种改进的U⁃net,选取交叉损失函数(constituent error,CE) 与Dice Loss 组合作为损失函数,提高网络模型对小目标的分类精度;添加批量归一化BN 层(batch normalization)模块和LayerScale 模块提高网络在训练时的收敛速度和稳定性,抑制无用特征并且突出有效特征;选取视觉几何组(visual geometry group, VGG)作为主干网络提高采样过程中对特征的重复利用率;选用两倍双线性插值上采样的方式,提高网络模型对特征的重复利用率。本文基于自制的单色图像数据集,利用改进的U⁃net 对单色图像进行语义分割,在像素准确率、类别平均像素准确率、平均交并比方面比原始的U⁃net 分别提高了1. 43%、1. 57%、8.75%,取得了良好的语义分割效果。

【关键字】单色图像;语义分割;改进U⁃net;交叉熵损失函数(CE);Dice Loss 损失函数;LayerScaler 模块
【PDF】