Table 2 Hyperparameters used in Mask R-CNN, DeepLabV3+, and SSD
From: Segmentation algorithm can be used for detecting hepatic fibrosis in SD rat
Mask R-CNN | DeepLabV3+ | SSD |
|---|---|---|
BACKBONE = resnet152 | BACKBONE = “resnet” | COCO_API = “PhythosAPI” |
DIST_BACKEND = “nccl” | LR = 0.01 | EPOCHS = 400 |
BATCH_SIZE = 16 | EPOCHS = 200 | BATCH_SIZE = 64 |
LR = 0.005 | BATCH_SIZE = 32 | LR = 0.00003 |
EPOCHS = 200 | DATASET = “coco” | WEIGHT_DECAY = 0.00003 |
WORKERS = 16 | MASK_THRESHOLD = 0.6 | MOMENTUM = 0.9 |
LR_STEPS = 16, 22 | CONFIDENCE = 0.5 | |
CONFIDENCE = 0.5 |