基于改进Mask R-CNN的水稻茎秆杂质分割方法研究
Research on segmentation method of rice stem impurities based on improved Mask R-CNN
-
摘要: 针对目前国内联合收割机缺乏含杂率在线检测的问题,提出一种基于改进Mask R-CNN的水稻茎秆杂质分割方法。依据茎秆杂质形状位置特征,对原始Mask R-CNN中网络层进行优化;引入图像增广技术对图像样本进行扩充,解决图像训练数据匮乏问题;利用训练后模型对验证集中图像进行分割,并与原始Mask R-CNN等算法进行对比。结果表明,改进后Mask R-CNN算法的综合评价指标F1达到91.12%,优于其他模型,且分割时间可达到3.57 s,证明其可满足实时检测要求,为后续含杂率在线检测系统实现提供技术参考。Abstract: Aiming at the current lack of online detection of impurity rate of domestic combine harvesters, a segmentation method for rice stems based on improved mask R-CNN was proposed. According to the shape and location characteristics of the stalk impurities, the network layer in the original Mask R-CNN was optimized. The image augmentation was introduced to expand the image samples and solve the lack of image training data. The trained model was used to segment the images in the verification set and compare them with the original Mask R-CNN and other algorithms. The results showed that the comprehensive evaluation index(F1) of the improved Mask R-CNN algorithm reaches 91.12%, which is better than other models. The segmentation time can reach 3.57 s, which proves that the method can meet the requirements of real-time detection and provides a technical reference for the subsequent implementation of an online detection system of impurity rate.