计算机应用

作者：贺杰安, 吴晓红, 何小海, 胡建蓉, 卿粼波

单位：1. 四川大学 电子信息学院, 成都 610065;2. 中储粮成都储藏研究院有限公司, 成都 610091

关键词：小麦不完善粒识别,卷积神经网络,L0平滑,图像增强,批正则化,分类

基金：四川省科技计划项目（2018HH0143）；四川省教育厅项目（18ZB0355）。

针对实际应用场景下，小麦籽粒图像背景单一以及小麦不完善粒的不完善特征大多是局部特征而大部分图像特征与正常粒无异的特点，提出一种基于细节的图像增强（IE）的小麦不完善粒识别方法。首先，使用交替最小化算法约束原图在水平方向和竖直方向的L0范数来平滑原图作为基础图层，并用原图减去基础图层得到图像的细节层；然后，突出细节层后将其与基础图层叠加以增强图像；最后，将增强后的图像作为卷积神经网络（CNN）的输入，使用加入了批正则化（BN）层的CNN对图像进行识别。分别以经典分类网络LeNet-5、ResNet-34、VGG-16和在其中添加BN层的这些网络作为分类网络，增强前后的图像作为输入来进行分类实验，并以测试集准确率评估性能。实验结果表明，三个经典分类网络均在添加了BN层后而使用相同输入时的测试集准确率提高了5个百分点，在使用细节增强后的图像作为输入时三个网络的测试集准确率提高了1个百分点，以上二者联合使用时三个网络均获得超过7个百分点的测试集准确率提升。

来源：2021年第3期

《计算机应用》期刊编辑部

查看计算机应用杂志2021年第3期