Skip to main content
. 2019 Dec 10;9:18738. doi: 10.1038/s41598-019-55373-7

Table 2.

Parameters used for training four different CNN-based algorithms for CAD.

Algorithm Mask R-CNN RetinaNet eDenseYOLO ASPP U-Net
Stages Two stages Single stage Single stage Single stage
Backbone Resnet101 ResNet101 DensNet201 ResNet50
Optimizer SGD SGD Adam Adam
Learning rate 1e-6 1e-6 1e-3 1e-3
Weight decay 0.0 0.0 0.0005
Momentum 0.9 0.9 0.9 0.9, 0.999