特征(语言学)
计算机科学
人工智能
特征提取
模式识别(心理学)
精确性和召回率
集合(抽象数据类型)
一般化
目标检测
放射性检测
算法
数学
数学分析
哲学
语言学
程序设计语言
作者
Haiying Liu,Xuehu Duan,Haonan Chen,Haitong Lou,Lixia Deng
摘要
Driven by deep learning, great breakthroughs had been made in the field of target detection. Small target detection algorithms were widely used in industry, agriculture and other fields. But the small target had few available features and the loss of small target detail information in feature extraction. So it led to the low accuracy of the small target detection algorithms. In this paper, we proposed DBF‐YOLO algorithm based on the classical YOLOV5. The classical YOLOV5 algorithm with high speed. The detection speed of the minimum model could reach 24 ms. However, the deep network structure led to the low detection accuracy of small targets. Our proposed DBF‐YOLO algorithm was an improvement on the problem of small target information being lost. The main contributions of this article were mainly: First, a shallow feature extraction network was introduced in P1 layer, more details of small targets could be well retained. Second, by adding the feature fusion network of shallow feature map and the detection output part in the FPN + PAN layers, the algorithm's accuracy and generalization ability were significantly enhanced. Compared to YOLOV5, the performance of the DBF‐YOLO algorithm was significantly improved. On the validation set, mAP@0.5 and mAP@0.5:0.95 were increased by 8.80 and 5.90%, respectively. Recall was increased from the initial 34.50–41.80%. Precision was increased from initial 44.20 to 50.70%. On the test set, mAP@0.5 and mAP@0.5:0.95 were increased by 6.40 and 3.90%, respectively. Recall was increased 5.10%. Precision was increased 6.60%. Experiments had shown that the improved algorithm achieved good results in accuracy. © 2023 Institute of Electrical Engineers of Japan. Published by Wiley Periodicals LLC.
科研通智能强力驱动
Strongly Powered by AbleSci AI