摘要
为实现中药饮片的高效分选,针对筛选过程中部分缺陷特征相似且难以区分的问题,提出了一种基于改进YOLOv5的中药饮片缺陷检测算法。首先,在Backbone中引入Faster Net网络结构,替换原始的C3结构,减少模型参数量,提高检测效率;其次,添加SimAM三维注意力模块,更好地提取目标特征;最后,引入Sim OTA标签匹配机制,提升模型训练速度的同时也提高检测精度。在黄芪饮片数据集上进行测试,最后结果表明,改进后的网络模型mAP为87.53%,相较于原始模型提高了1.78%,对中药饮片各类缺陷识别能力更强。
To achieve efficient sorting of Chinese herbal medicine pieces,and to address the issue of some defect features being similar and difficult to distinguish during the screening process,this paper proposes a Chinese herbal medicine piece defect detection algorithm based on an improved YOLOv5.Firstly,the Faster Net network structure is introduced in the Backbone,replacing the original C3 structure,to reduce the model parameter count and improve detection efficiency.Secondly,a SimAM three-dimensional attention module is added to better extract target features.Lastly,the Sim OTA label matching mechanism is introduced to increase the training speed of the model while also enhancing detection accuracy.Testing on the Astragalus herbal pieces dataset,the final results show that the improved network model achieves a mean Average Precision of 87.53%,which is a 1.78% improvement over the original model,indicating a stronger capability in recognizing various defects in Chinese herbal medicine pieces.
作者
李云阳
李根
闫磊
LI Yun-yang;LI Gen;YAN Lei(School of Technology,Beijing Forestry University,Beijing 100083,China)
出处
《林业机械与木工设备》
2024年第5期71-75,82,共6页
Forestry Machinery & Woodworking Equipment
基金
国家自然科学基金面上项目(31770769)。
