README.md

November 29, 2021 ยท View on GitHub

Transferring to Detection

The train_object_detection.py script reproduces the object detection experiments on Pascal VOC and COCO.

Instruction

  1. Install detectron2.

  2. Convert a pre-trained model to detectron2's format:

    python3 convert_model_to_detectron2.py --pretrained_feature_extractor PATH_TO_CKPT --output_detectron_model ./detectron_model.pkl

  3. Put dataset under "./datasets" directory, following the directory structure requried by detectron2.

  4. Run training:

    python train_net.py --config-file configs/pascal_voc_R_50_C4_24k_moco.yaml \
 --num-gpus 8 MODEL.WEIGHTS ./detectron_model.pkl