Table 2:
Localization network | Segmentation network | |
---|---|---|
Trainable parameters | 145k | 1.1M |
Epochs | 12 | 12 |
Optimizer | Adam | Adam |
learning rate | 0.001 | 0.001 |
Batch size | 8 | 1 |
Loss | MSE | DiceCE |
Abbreviation: MSE: Mean Squared Error loss; DiceCE: Soft dice loss + Cross Entropy loss.