Skip to content

Commit c05da2a

Browse files
yoshitomo-matsubarafmassa
authored andcommitted
add a README for training object detection models (#1612)
1 parent 537f0df commit c05da2a

File tree

1 file changed

+31
-0
lines changed

1 file changed

+31
-0
lines changed

references/detection/README.md

Lines changed: 31 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,31 @@
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+

0 commit comments

Comments
 (0)