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
45 stars 6 forks source link

Few shot or Zero shot? #21

Open tungyen opened 1 week ago

tungyen commented 1 week ago

Hi, thanks for this amazing work. I want to ask if this model is few-shot or zero-shot model for 6D pose estimation? Thank you!

cviviers commented 6 days ago

Hi @tungyen It is many shot I guess. You have to train it on your specific object. The objective is more on the speed and accuracy. Nvidia has a nice FoundationPose model for zero-shot pose estimation