From c05da2a84c1239a9d57431cec7e9ed83931d478b Mon Sep 17 00:00:00 2001 From: Yoshitomo Matsubara Date: Tue, 26 Nov 2019 02:52:23 -0800 Subject: [PATCH] add a README for training object detection models (#1612) --- references/detection/README.md | 31 +++++++++++++++++++++++++++++++ 1 file changed, 31 insertions(+) create mode 100644 references/detection/README.md diff --git a/references/detection/README.md b/references/detection/README.md new file mode 100644 index 00000000000..7aec36aefa3 --- /dev/null +++ b/references/detection/README.md @@ -0,0 +1,31 @@ +# Object detection reference training scripts + +This folder contains reference training scripts for object detection. +They serve as a log of how to train specific models, as provide baseline +training and evaluation scripts to quickly bootstrap research. + +Except otherwise noted, all models have been trained on 8x V100 GPUs. + +### Faster R-CNN +``` +python -m torch.distributed.launch --nproc_per_node=8 --use_env train.py\ + --dataset coco --model fasterrcnn_resnet50_fpn --epochs 26\ + --lr-steps 16 22 --aspect-ratio-group-factor 3 +``` + + +### Mask R-CNN +``` +python -m torch.distributed.launch --nproc_per_node=8 --use_env train.py\ + --dataset coco --model maskrcnn_resnet50_fpn --epochs 26\ + --lr-steps 16 22 --aspect-ratio-group-factor 3 +``` + + +### Keypoint R-CNN +``` +python -m torch.distributed.launch --nproc_per_node=8 --use_env train.py\ + --dataset coco_kp --model keypointrcnn_resnet50_fpn --epochs 46\ + --lr-steps 36 43 --aspect-ratio-group-factor 3 +``` +