基于深度学习的粘连米粒实例分割算法研究
Research on Instance Segmentation Algorithm of Adhesive Rice Grains Based on Deep Learning作者机构:浙江工商大学信息与电子工程学院杭州310018
出 版 物:《中国粮油学报》 (Journal of the Chinese Cereals and Oils Association)
年 卷 期:2022年第37卷第9期
页 面:224-230页
核心收录:
学科分类:08[工学] 081101[工学-控制理论与控制工程] 0811[工学-控制科学与工程] 081102[工学-检测技术与自动化装置]
摘 要:基于机器视觉的农作物外观质量检测近些年越来越受到人们的关注。在抽样检测台上由于米粒可能相互触碰、粘连,采集的图像若不进行分割预处理会造成后续大米外观品质评测的失误。由此,本实验提出了一种基于Mask R-CNN改进的实例分割网络RiceInstNet,用于粘连米粒的图像分割。主干网络由2个改进的VoVNetV2网络并行组成,在大规模减少网络参数的同时加强对粘连米粒图像的特征提取,另外在掩膜分支上增加了一个学习物体边界的子网络,利用边界特征丰富掩膜特征并促进掩膜预测得更加精细。实验结果表明,该网络可以实现对粘连米粒的实例分割,与Mask R-CNN网络相比,改进后的网络模型RiceInstNet的平均精度和召回率分别由87.2%、89.7%提升到90.4%、93.7%,同时本网络模型更轻量,非常适合集成到移动终端或嵌入式设备中。