Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Training model on REAL275 dataset #4

Open
PnthrLeo opened this issue Feb 26, 2022 · 4 comments
Open

Training model on REAL275 dataset #4

PnthrLeo opened this issue Feb 26, 2022 · 4 comments

Comments

@PnthrLeo
Copy link

PnthrLeo commented Feb 26, 2022

Recently, I have tried to reproduce results that was given in the DualPoseNet paper. Therefore, I used training_data_prepare.py script to generate necessary training data (for CAMERA25 and REAL275 datasets), which allowed me successfully reproduce results on CAMERA25 dataset.

However, results after training model on REAL275 were very far from the paper results. So, after that I began to wonder -- maybe, I do something wrong in case of training/evaluation on REAL275 dataset.

My pipeline of actions for REAL275 was:

  1. Download NOCS data and run python training_data_prepare.py.
  2. Run python main.py --phase train --dataset REAL275
  3. Run python main.py --phase test --dataset REAL275

I did not save results of mine model evaluation on REAL275, but if you need it I can recalculate them for the issue investigation!

Thank you in advance!

@PnthrLeo
Copy link
Author

PnthrLeo commented Feb 28, 2022

Oh, I guess, I found where was the problem. Last time I trained model for REAL275 dataset without CAMERA25_*.pkl files. So this time I will try to put CAMERA25_*.pkl files with REAL275.pkl file together and run the training pipeline on REAL275 dataset again. I will write about my new results as soon as I calculate them!

@PnthrLeo
Copy link
Author

In the end, seems like my last suggestion was correct (I achieved results close to the one shown in the original paper).

@Ysh1n3
Copy link

Ysh1n3 commented Apr 10, 2023

Hello, may I know what environment you are using to implement this code?

@Ysh1n3
Copy link

Ysh1n3 commented Apr 10, 2023

I am using the same environment as the author, but I cannot implement this error.

libcublas.so.9.0: cannot open shared object file: No such file or directory

I am using RTX 3060

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants