Research Article

A Robust and Lightweight Detector for Ship Target with Complex Background in SAR Image

Algorithm 1

3S-YOLO detection algorithm.
Preliminary work : data set input, training parameter setting, training period epoch
Pruning rate : R
While(FPGM pruning model does not converge or T<epoch)
1.Backbone
   Feature information was extracted using Focus, C3 and CBL structures.
        
        
2.Neck
   Scale change using up-sampling and down-sampling, feature fusion using Concat.
   Computational fusion features :
        
3. Prediction layer
   Use Varifocal loss to highlight positive sample weights. EIoU anchor frame positioning.
   EIoU loss:
       
   Varifocal-EIoU:
     
End While
Network compression, redundant parameter elimination after pruning.
Output : pruning and model training convergence network model and weight