The code is based on Dectectron2.

We used colab to implement algorithms at first. We tested the performance of pre-trained models given by Detectron2 in Task1&2.ipynb for both Task 1 and Task 2.

retrain.py is for retraining.

inference.py is for inference.

For Task1&2.ipynb, you need to have a jupyter notebook environment.

For retrain.py and inference.py, you need to install Dectectron2 first following the instructions in Task1&2.ipynb. Then use the command line:

python retrain.py
python inference.py