Label Assignment Algorithm for Metal Defect Samples Based on Distance-awareness
The distance-aware dynamic label assignment(DDA)algorithm was proposed to address issues such as the lack of consideration for aspect ratio of metal surface defects and poor localization ability towards target distribution during the allocation of positive and negative samples in training processes of metal surface defect detection model.DDA did not change the structures of original detec-tion model and did not increase computational expenses.A new distance loss calculation paradigm was proposed based on geometric characteristics of real frame to optimize the regression problem with a wide aspect ratio.The regression offset in iterative processes was decoded as predicted frame coordi-nates.Finally,the comprehensive intersection and union ratio information were calculated among the predicted frame,anchor frame and real frame,and positive and negative samples were dynamically se-lected to improve training accuracy.It was verified through the surface defect detection task of cold-rolled strip in a steel plant in Wuhan,and a public hot-rolled strip surface defect data set was intro-duced for generalization testing.The detection results are significantly improved,which has practical application values for metal surface quality specifications.
object detectionsample selection strategyaspect ratiometal defect detectiondis-tance regression loss function
国家科技期刊平台
NSTL
万方数据
