Where is the inference pt file?

cviviers / YOLOv5-6D-Pose

6-DoF Pose estimation based on the YOLOv5 framework. Specific focus on instruments in X-ray applications

https://ieeexplore.ieee.org/document/10478293

GNU General Public License v3.0

32 stars 6 forks source link

Where is the inference pt file? #11

Closed panwenfeics26 closed 1 week ago

panwenfeics26 commented 2 weeks ago

python detect.py --weights path/to/weights/linemod/cat/best.pt --img 640 --conf 0.25 --source ../data/LINEMOD/cat/JPEGImages --static-camera configs/linemod/linemod_camera.json --mesh-data path/to/data/LINEMOD/cat/cat.ply

I can't find where best.pt is. What are the linemod_weights downloaded?

cviviers commented 2 weeks ago

Hey @panwenfeics26 ,

You can download all the weights here. Please follow the detailed tutorial on how to set up the experiments.