MMDetection

Install

pip install -r requirements.txt -f https://download.openmmlab.com/mmcv/dist/{cuda_version}/{torch_version}/index.html

Example:

pip install -r requirements.txt -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.10.0/index.html

Dataset

python3 data/download_dataset.py --dataset-name coco2017 --save-dir data/coco --unzip --delete

Train

Double Head

python train/train.py /content/MMDetection/configs/double_heads/dh_faster_rcnn_r50_fpn_1x_coco.py \
                      --work-dir train/double_head_train

Double Head Extension

python train/train.py ./configs/double_heads/dh_ext_faster_rcnn_r50_fpn_1x_coco.py \
                      --work-dir train/double_head_train

Mask RCNN

python train/train.py /content/MMDetection/configs/mask_rcnn/mask_rcnn_r50_fpn_2x_coco.py \
                      --work-dir train/mask_rcnn_train

Evaluation

!python test/evaluation.py /content/MMDetection/configs/double_heads/dh_faster_rcnn_r50_fpn_1x_coco.py \
                          train/work_dirs/latest.pth \
                          --work-dir test/work_dirs \
                          --eval mAP bbox