Defect detection of injection-molded parts based on improved-YOLOv5

Quality inspection with injection-molded parts (IMPs) is an essential part of the process of producing IMPs. To improve production quality, we propose an improved-YOLOv5 model to detect the object defection of the IMPs. Different from the original YOLOv5, we replace the coupled detection head (CH) with the independent decoupled head (IDCH), while adding the Squeeze-and-Excitation attention mechanism (SE) into the regression branch. Furthermore, a defective dataset of IMPs is constructed within the real-world production line, including 1357 IMPs images. The experimental results demonstrate the effectiveness of the improve-YOLOv5. Compared with YOLOv5, the precision and recall of the modified algorithm in this paper are increased by 2.5 % and 4.7 %, respectively. For difficult categories, precision and recall are increased by 8.7 % and 13.5 %, respectively.