zju3dv / OnePose

Code for "OnePose: One-Shot Object Pose Estimation without CAD Models", CVPR 2022
Apache License 2.0
935 stars 78 forks source link

training on custom data #51

Open rhalder2023 opened 1 year ago

rhalder2023 commented 1 year ago

Hi, so i am trying to use your model to train a custom dataset to calculate 6d poses. Firstly, i used your app to get annotated data, but when i am trying to use the testing data capture mode, your app OnePose cap is not sending me a link to download the testing data.

rhalder2023 commented 1 year ago

Also I wanted to know is it a must we use the cuda and pytorch version thats been mentioned in the environment.yaml file or can we make it work also with the latest releases for both

Mechazo11 commented 2 months ago

@rhalder2023 Hi, I too am looking to train a custom dataset for this paper. Can you add some details regarding the pipeline you are using?