Injection molded parts are increasingly utilized across various industries due to their cost-effectiveness, lightweight nature, and durability. However, traditional defect detection methods for these parts often rely on manual visual inspection, which is inefficient, expensive, and prone to errors. To enhance the accuracy of defect detection in injection molded parts, a new method called MRB-YOLO, based on the YOLOv8 model, has been proposed. This method introduces several key improvements: (1) the MAFHead, a four-detection head based on multiplicative feature fusion, which replaces the original detection head to enhance feature representation; (2) the RepGFPN-SE module, a re-parameterized generalized feature pyramid network that improves detection of small objects by replacing the original C2f. module; (3) and the BiNorma module, employing a bi-level routing attention mechanism to optimize the training process by reducing input distribution changes across layers. The effectiveness of the MRB-YOLO model was validated through ablation and contrast experiments using a specially constructed dataset of injection molded parts defects. The results demonstrated an accuracy of 88.8%, a recall rate of 86.8%, and a mean average precision (mAP) of 91.5%. Compared to the YOLOv8n model, the MRB-YOLO model shows an increase in accuracy by 8.2%, in recall rate by 17.2%, and in mAP by 11.8%. These findings confirm that the MRB-YOLO model meets the requirements for accurate detection of defects in injection molded parts.
Read full abstract