How to test the pre-trained model?

srijandas07 / VPN

Pose driven attention mechanism

37 stars 9 forks source link

How to test the pre-trained model? #1

Open MuyangLi-Dawnlight opened 3 years ago

MuyangLi-Dawnlight commented 3 years ago

Hi, I am wondering how to test the pre-trained models, especially on the smarthome dataset?

srijandas07 commented 3 years ago

@MuyangLi-Dawnlight, the test code is not so different. Follow the training code, just load the model weights, and use model.predict_generator. Some test codes are provided in my other repositories. The test code for this repo will be updated after Nov 16.