基于U<sup>2</sup>⁃Net的金属表面缺陷检测算法

基于U²⁃Net的金属表面缺陷检测算法

王昱翔, 葛洪伟

Metal surface defect detection algorithm based on U²⁃Net

Yuxiang Wang, Hongwei Ge

表1 本文方法同其他六种方法在KolektorSDD数据集上的性能比较

Table 1 Performance of our method and six other methods on KolektorSDD dataset

	Acc	P	R	F₁⁃score	F₂⁃score	MAE	MSE
本文方法	0.998	0.794	0.843	0.812	0.833	0.277	0.001
U⁃Net	0.977	0.682	0.789	0.732	0.850	10.81	50.63
Att⁃Unet	0.987	0.762	0.812	0.759	0.790	10.36	30.88
U²⁃Net	0.982	0.763	0.809	0.785	0.799	0.858	0.794
Nested⁃UNet	0.989	0.737	0.823	0.778	0.804	10.10	10.23
Res⁃ Unet++	0.993	0.774	0.805	0.789	0.798	0.889	0.908
R2U_Net	0.992	0.761	0.813	0.786	0.802	10.01	10.28