Skip to main content

Table 1 Training parameters for Mask R-CNN, MS R-CNN, D2Det and YOLACT

From: Examination of blood samples using deep learning and mobile microscopy

Framework

# epoch

Learning rate

Batch size

Weight decay

Basis for pretrained weights

Original

Modified

Original

Modified

Original

Modified

Mask R-CNN

12

1000

0.02

0.02

2

2

0.0001

COCO [57]

MS R-CNN

3750*

1000*

0.02

0.00025

16

1

0.0001

ImageNet [58]

D2Det

24

3000

0.02

0.02

2

2

0.0001

COCO [59]

YOLACT

66.666*

1000–2000

0.0001

0.0001

8

2

0.0005

COCO [60]

  1. *Calculated by: # epoch = number of iterations*batch size/number of training images