How to test your trained model on my own dataset?

[Drow999]

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]

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.