File tree Expand file tree Collapse file tree 1 file changed +31
-0
lines changed Expand file tree Collapse file tree 1 file changed +31
-0
lines changed Original file line number Diff line number Diff line change
1
+ # Object detection reference training scripts
2
+
3
+ This folder contains reference training scripts for object detection.
4
+ They serve as a log of how to train specific models, as provide baseline
5
+ training and evaluation scripts to quickly bootstrap research.
6
+
7
+ Except otherwise noted, all models have been trained on 8x V100 GPUs.
8
+
9
+ ### Faster R-CNN
10
+ ```
11
+ python -m torch.distributed.launch --nproc_per_node=8 --use_env train.py\
12
+ --dataset coco --model fasterrcnn_resnet50_fpn --epochs 26\
13
+ --lr-steps 16 22 --aspect-ratio-group-factor 3
14
+ ```
15
+
16
+
17
+ ### Mask R-CNN
18
+ ```
19
+ python -m torch.distributed.launch --nproc_per_node=8 --use_env train.py\
20
+ --dataset coco --model maskrcnn_resnet50_fpn --epochs 26\
21
+ --lr-steps 16 22 --aspect-ratio-group-factor 3
22
+ ```
23
+
24
+
25
+ ### Keypoint R-CNN
26
+ ```
27
+ python -m torch.distributed.launch --nproc_per_node=8 --use_env train.py\
28
+ --dataset coco_kp --model keypointrcnn_resnet50_fpn --epochs 46\
29
+ --lr-steps 36 43 --aspect-ratio-group-factor 3
30
+ ```
31
+
You can’t perform that action at this time.
0 commit comments