HTC reimplementation using PyTorch (supports MOSAIC and MixUp)

Train (default configuration is for COCO dataset)

• Install mmdet toolbox
• Register your dataset to utils/dataset.py, see INSDataset
• See nets/exp01.py for using MOSAIC and MixUp data pipeline
• See nets/exp02.py for using RandomAugment data pipeline
• Run bash ./main.sh ./nets/exp01.py $ --train for training, $ is number of GPUs

Reference

• https://github.com/ultralytics/yolov5
• https://github.com/open-mmlab/mmdetection