vietnh1009 / SSD-pytorch

SSD: Single Shot MultiBox Detector pytorch implementation focusing on simplicity
MIT License
164 stars 49 forks source link

export the coco predictions in json file #4

Closed 123dddd closed 3 years ago

123dddd commented 3 years ago

Hello Viet, thanks for this repo! I noticed that you used the coco eval API to evaluate the the trained model and the results are clear and impressive. I am wondering if the coco eval API also can be used in the test_dataset.py to evaluate the prediction results on coco dataset, such as coco eval2017 dataset? Any suggestion will be appreciated ! : )