The recall rate of the original YOLOv4 model for detecting internal defects in aluminum alloy welds is relatively low.To address this issue, this Pillow Block Bearing paper introduces an enhanced model, YOLOv4-cs1.The improvements include optimizing the stacking method of residual blocks, modifying the activation functions for different convolutional layers, and eliminating the downsampling layer in the PANet (Pyramid Attention Network) to preserve edge information.Building on these enhancements, the YOLOv4-cs2 model further incorporates an improved Spatial Pyramid Pooling (SPP) module Strap-ons after the third and fourth residual blocks.
The experimental results demonstrate that the recall rates for pore and slag inclusion detection using the YOLOv4-cs1 and YOLOv4-cs2 models increased by 28.9% and 16.6%, and 45% and 25.2%, respectively, compared to the original YOLOv4 model.
Additionally, the mAP values for the two models are 85.79% and 87.5%, representing increases of 0.98% and 2.
69%, respectively, over the original YOLOv4 model.