tools

Requirements:

• PyTorch >=1.4
• torchvision from master
• cocoapi
• yacs
• matplotlib
• GCC >= 4.9
• detectron2

Our model are mainly based on detectron2 architecture. The main modifications are at meta_arch and backbone. User can replace the original part with ours.

Training

python train.py

Test

python eval_my.py