无线电工程2024，Vol.54Issue(6)：1446-1453,8.DOI:10.3969/j.issn.1003-3106.2024.06.013
PSOS-YOLOv5s:一种轻量级玉米雄穗快速检测算法
PSOS-YOLOv5s:A Lightweight Maize Tassel Rapid Detection Algorithm
摘要
Abstract
In order to solve the problem of slow speed in maize tassel detection,a lightweight and fast detection algorithm PSOS-YOLOv5s based on YOLOv5s improvement is proposed.The proposed algorithm uses lightweight PP-LCNet to replace CSPDarknet53 in the backbone network,and uses a combination of two different depth-separable convolutions and attention mechanisms to form basic blocks,reducing model complexity and speeding up detection;in the prediction part,the SimOTA label matching strategy is used to replace the label matching strategy in YOLOv5s,the central prior idea is used to obtain accurate prior knowledge,and the dynamic k strategy is proposed to filter redundant labels to improve the model's ability to quickly select positive samples;SIOU loss is used in the prediction part to replace GIOU Loss and the angle loss factor is introduced to reduce the degree of freedom of regression,speed up the convergence speed,save training time,and redefine the penalty index to improve detection accuracy.Experimental results show that in the maize tassel data set,compared with YOLOv5s,the number of model parameters of the proposed improved algorithm is reduced by 52.86％,the detection accuracy of the model is increased by 0.6％,and the detection speed of the model is increased by 65.5％.It shows that the improved algorithm has obvious improvement effect and can meet the requirements of rapid detection of large-scale maize tassels.关键词
玉米雄穗检测/轻量化网络/标签匹配策略/损失函数/YOLOv5Key words
maize tassel detection/lightweight network/label matching strategy/loss function/YOLOv5分类
信息技术与安全科学引用本文复制引用
胡阵,马宗军,黄传宝,赵景波,唐勇伟,郝凤琦..PSOS-YOLOv5s:一种轻量级玉米雄穗快速检测算法[J].无线电工程,2024,54(6):1446-1453,8.基金项目
山东省科技型中小企业创新能力提升工程项目(2023TSGC0111,2023TSGC0587) （2023TSGC0111,2023TSGC0587）
青岛市民生计划(22-3-7-xdny-18-nsh) （22-3-7-xdny-18-nsh）
山东省重点研发计划(软科学项目)(2023RZA02017) （软科学项目）
山东省重大科技创新工程项目(2019JZZY020603)Innovation Capability Improvement Project for Small and Medium-sized Scientific and Technological Enterprises in Shandong Province(2023TSGC0111,2023TSGC0587) （2019JZZY020603）
Qingdao People's Livelihood Planning(22-3-7-xdny-18-nsh) （22-3-7-xdny-18-nsh）
Key Research and Development Plan of Shandong Prov-ince(Soft Science Project)(2023RZA02017) （Soft Science Project）
Major Science and Technology Innovation Project of Shandong Province(2019JZZY020603) （2019JZZY020603）
