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

How to test your trained model on my own dataset? #10

Open
Drow999 opened this issue Dec 11, 2021 · 1 comment
Open

How to test your trained model on my own dataset? #10

Drow999 opened this issue Dec 11, 2021 · 1 comment

Comments

@Drow999
Copy link

Drow999 commented Dec 11, 2021

Hello,
Could you tell me how to test your model on my own dataset? And I don't want to train the model again.
I created a directory real_animal_all in ./animal_data/ and put a few images in it, and I run the code but there is no response
python evaluation/test.py --arch pose_resnet_refine_mt_multida --animal all --dataset real_animal_all --resume checkpoint/all/models/stage2/model_best.pth.tar --evaluate --train_on_all_cat --flip
Is this the right way to test new data?

@chaneyddtt
Copy link
Owner

Hi @Drow999, you might need to write a dataloader for your own data and then test our model. Alternatively, you can modify the dataloader under the pose/datasets/ folder, for example real_animal_all.py, to load your own images.

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