Multi-Scale Target Detection Algorithm Based on Attention Mechanism

Moran, 鞠默然 Ju; Jiangning, 罗江宁 Luo; Zhongbo, 王仲博 Wang; Haibo, 罗海波 Luo

doi:10.3788/aos202040.1315002

Cited by 8 publications

(1 citation statement)

References 0 publications

Supporting

Mentioning

Contrasting

Order By: Relevance

“…The swift progress in computer networks has led to the proposal of several feature fusion methods [29][30][31][32][33]. These methods aim to merge the detailed location information from shallow feature maps with the comprehensive semantic insights from deeper layers.…”

Section: Feature Fusion Strategymentioning

confidence: 99%

Automotive Parts Defect Detection Based on YOLOv7

Huang,

Zhu

2024

Electronics

View full text Add to dashboard Cite

Various complex defects can occur on the surfaces of small automobile parts during manufacturing. Compared with other datasets, the auto parts defect dataset used in this paper has low detection accuracy due to various defects with large size differences, and traditional target detection algorithms have been proven to be ineffective, which often leads to missing detection or wrong identification. To address these issues, this paper introduces a defect detection algorithm based on YOLOv7. To enhance the detection of small objects and streamline the model, we incorporate the ECA attention mechanism into the network structure’s backbone. Considering the small sizes of defect targets on automotive parts and the complexity of their backgrounds, we redesign the neck portion of the model. This redesign includes the integration of the BiFPN feature fusion module to enhance feature fusion, with the aim of minimizing missed detections and false alarms. Additionally, we employ the Alpha-IoU loss function in the prediction phase to enhance the model’s accuracy, which is crucial for reducing false detection. The IoU loss function also boosts the model’s efficiency at converging. The evaluation of this model utilized the Northeastern University steel dataset and a proprietary dataset and demonstrated that the average accuracy (mAP) of the MBEA-YOLOv7 detection network was 76.2% and 94.1%, respectively. These figures represent improvements of 5.7% and 4.7% over the original YOLOv7 network. Moreover, the detection speed for individual images ranges between 1–2 ms. This enhancement in detection accuracy for small targets does not compromise detection speed, fulfilling the requirements for real-time, dynamic inspection of defects.

show abstract

Section: Feature Fusion Strategymentioning

confidence: 99%