TY - GEN
T1 - Research and application based on the improved YOLO V7 target detection algorithm
AU - Wang, Tenghui
AU - Zhang, Xiaofeng
AU - Ma, Yan
AU - Wang, Yapeng
AU - Xie, Haijun
AU - Zhu, Mingchao
AU - Su, Binghua
AU - Yao, Dong
N1 - Publisher Copyright:
© 2024 SPIE.
PY - 2024
Y1 - 2024
N2 - At present, YOLO-based of algorithms have been widely used in urban planning, traffic monitoring, ecological protection, military security and other fields, and their applicable scenarios are expanding. In view of the problems of high misdetection rate, high omission rate and insufficient accuracy in the image target detection task, this topic is dedicated to study the improved target detection algorithm based on YOLO V7. In order to optimize the time cost and computing resource consumption, the Anchor-free based design is introduced, and through optimizing the design of decoupling head, the independent processing of classification and regression tasks is realized to improve the efficiency of feature extraction. Based on this method, the CBATM attention mechanism is used to better capture the intercorrelations between features and improve the representational ability of the model. In the loss function section, this paper adopts the SimOTA method in YOLOX to realize the dynamic number allocation of positive samples, which greatly reduces the training time. Improved YOLO V7 target detection algorithm in the PASCAL VOC challenge public dataset VOC2007 data set, the results show that the improved YOLO V7 target detection algorithm than the original YOLO V7, has higher detection accuracy and efficient performance, the average detection accuracy (mAP) increased by 2.27%, compared with other classic target detection algorithm, the improved YOLO V7 performance is more accurate and efficient.
AB - At present, YOLO-based of algorithms have been widely used in urban planning, traffic monitoring, ecological protection, military security and other fields, and their applicable scenarios are expanding. In view of the problems of high misdetection rate, high omission rate and insufficient accuracy in the image target detection task, this topic is dedicated to study the improved target detection algorithm based on YOLO V7. In order to optimize the time cost and computing resource consumption, the Anchor-free based design is introduced, and through optimizing the design of decoupling head, the independent processing of classification and regression tasks is realized to improve the efficiency of feature extraction. Based on this method, the CBATM attention mechanism is used to better capture the intercorrelations between features and improve the representational ability of the model. In the loss function section, this paper adopts the SimOTA method in YOLOX to realize the dynamic number allocation of positive samples, which greatly reduces the training time. Improved YOLO V7 target detection algorithm in the PASCAL VOC challenge public dataset VOC2007 data set, the results show that the improved YOLO V7 target detection algorithm than the original YOLO V7, has higher detection accuracy and efficient performance, the average detection accuracy (mAP) increased by 2.27%, compared with other classic target detection algorithm, the improved YOLO V7 performance is more accurate and efficient.
KW - Anchor-free
KW - CBAM attention mechanism
KW - Decoupled Head
KW - YOLO V7
UR - http://www.scopus.com/inward/record.url?scp=85208826572&partnerID=8YFLogxK
U2 - 10.1117/12.3050416
DO - 10.1117/12.3050416
M3 - Conference contribution
AN - SCOPUS:85208826572
T3 - Proceedings of SPIE - The International Society for Optical Engineering
BT - Third International Conference on Image Processing, Object Detection, and Tracking, IPODT 2024
A2 - Liu, Bin
A2 - Leng, Lu
PB - SPIE
T2 - 2024 3rd International Conference on Image Processing, Object Detection, and Tracking, IPODT 2024
Y2 - 9 August 2024 through 11 August 2024
ER -