Table 3. Comparison experiments.
Model | Dice | Training time | Inference time | Params |
---|---|---|---|---|
FCN | 0.61 | 2.17h | 0.045s | 270M |
UNet | 0.75 | 2.50h | 0.022s | 33M |
UNet++ | 0.85 | 1.49h | 0.029s | 30M |
LinkNet | 0.79 | 1.21h | 0.030s | 12M |
DeepLabV3 | 0.81 | 2.30h | 0.030s | 54M |
DeepLabV3+ | 0.85 | 1.80h | 0.025s | 41M |
UNet++ (with CA) |
0.86 | 1.52h | 0.030s | 31M |
UNet++ (with MSS) |
0.88 | 1.88h | 0.039s | 33M |
CM-UNet | 0.89 | 1.90h | 0.039s | 33M |
Comparison experiment of our segmentation algorithm with other classical segmentation algorithms. The inference time is calculated by the single image.