This implementation is adapted from torchvision object detection tutorial
sudo pip3 install -r requirements.txt
- Data preparation: data.py
- Model: model.py
- Training: eureka_training.py
- Inference: eureka_infer.py
- Visualization: visualize.py
- Others: coco_eval.py, coco_utils.py, engine.py, utils.py, and transforms.py are copied from https://github.com/pytorch/vision/blob/master/references/detection/