- 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.
python train.py
python eval_my.py