摘要
为了解决烟叶原料中烟草甲人工识别效率低、不准确的问题,实现烟草甲的精准、智能识别,提出一种基于YOLOv3改进算法的烟草甲识别模型。首先采用Random Mix数据增强技术扩充烟草甲数据集;然后,通过引入K-means++算法重新聚类锚框,改进YOLOv3对于烟草甲的识别能力;使用SIoU Loss改进YOLOv3的边界框损失函数,提高模型定位的准确性,加速模型收敛;最后加入特征细化模块过滤冲突信息,优化模型对烟草甲数据集中小目标识别的精准度。结果表明,改进的YOLOv3网络模型对测试集中烟草甲的平均检测精确率、召回率、F1分数和平均精度均值(mAP)分别达到93.26%、88.25%、0.90和94.59%,分别比现有的YOLOv3算法提高了12.21百分点、11.79百分点、0.12、12.40百分点,可为烟草甲的有效识别提供支撑。
-
单位郑州轻工业大学; 浙江中烟工业有限责任公司