Table 5.
Comparison of models trained with binary cross-entropy and dice coefficient loss. The model performance is reported as: the center deviation (Cen Dev), scale deviation (Sca Dev), 2D IOU and 3D IOU. The model trained with dice coeffcient loss outperformed the model trained with binary cross-entropy.
| Loss | Cen Dev (mm) | Sca Dev (mm) | 2D IOU (%) | 3D IOU (%) |
|---|---|---|---|---|
| Cross-entropy | 2.0 ± 1.9 | 2.2 ± 1.8 | 71.7 ± 8.1 | 62.0 ± 15.8 |
| Dice | 1.8 ± 1.4 | 1.9 ± 1.5 | 74.6 ± 7.1 | 63.2 ± 14.7 |