Model | Resolution | Epochs | LR | Batch | Optimizer | Loss |
HRNet | 1280 × 1280 | 80,000 | 1 × 10−4 | 2 | Adam | Cross Entropy |
Resnet (both 18 & 34) | 243 × 324 | Frozen 15, Unfrozen 30 | LR Finder | 12 | Adam | Cross Entropy |
Resnet (both 18 & 34) | 486 × 648 | Frozen 15, Unfrozen 30 | LR Finder | 4 | Adam | Cross Entropy |
Resnet (both 18 & 34) | 972 × 1296 | Frozen 15, Unfrozen 40 | LR Finder | 1 | Adam | Cross Entropy |
Efficientnet | 1280 × 1280 | 13 | 1 × 10−4 | 2 | Adam | Cross Entropy |
Detectron2 | 2592 × 1944 | 3000 | 25 × 10−5 | 8 | SGD | Cross Entropy |