search

MohsenZand / ObjectBox

(ECCV 22 Oral) ObjectBox: From Centers to Boxes for Anchor-Free Object Detection
GNU General Public License v3.0
131 stars 16 forks source link

How do I load the .pth weight file? #11

Closed 1358422654 closed 1 year ago

MohsenZand commented 1 year ago

In the 'flag_sets.py', set the 'weights' flag as the path to the checkpoint file (best.pt). Also, the 'task' flag has to be changed properly.