valerija-h / os_tog

Code for RA-L paper "One-shot Learning for Task-oriented Grasping"
3 stars 1 forks source link

Inferencing #2

Open Sai-Yarlagadda opened 1 month ago

valerija-h commented 1 month ago

If you want to make it work for new objects/task categories you will need to build your own OSTOG_physical_experiments.json as the one-shot learning models use it for references. Depending on how comfortable you are with transforming masks you can use MakeSense.AI or CVAT for labeling your own images, which are both free to use.