jahongir7174 / DualHTC

DualHTC++ implementation using PyTorch (supports MOSAIC/MixUp and RandomAugment)

DualHTC++ implementation using PyTorch (supports MOSAIC and MixUp)

Train

• 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

Note

• The default configuration is for 2021 VIPriors Instance Segmentation Challenge dataset

Reference

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