Scripts to prepare training dataset are listed in training_dataset directory.
Download pretrained backbones from google driver or BaiduYun (code: 5o1d) and put them into pretrained_models directory.
cd experiments/train/got10k
cd experiments/train/fulldata
CUDA_VISIBLE_DEVICES=0,1,2,3
python -m torch.distributed.launch
--nproc_per_node=4
--master_port=2333
../../../tools/train.py --cfg config.yaml
The code is implemented based on RBO. We would like to express our sincere thanks to the contributor.