3.7.12
pytorch=1.10.0
torchvision=0.11.1
h5py
ptqdm
seaborn
opencv
openmim
mvcc-full
https://github.com/open-mmlab/mmdetection
https://github.com/ultralytics/yolov5
Download Link Colab
or
from inference import inference
'''
assume folder structure
./
|model.pt
|data/
|test/... << test image folder
|yolov5/... << git cloned yolov5
'''
inference('./model.pt', './data/test', './answer.json') # model_path, test_path, output_path