search

phamquiluan / ResidualMaskingNetwork

ICPR 2020: Facial Expression Recognition using Residual Masking Network
https://ieeexplore.ieee.org/document/9411919
MIT License
448 stars 91 forks source link

How to get the model once it has been trained #50

Closed Daniel-CG-Wright closed 7 months ago

Daniel-CG-Wright commented 7 months ago

Hi, I have trained the model by running main_fer2013.py and it successfully did the training (after adding the fer dataset to saved/ etc) However, how can I then use the model to predict images once the model is trained? I see it created a file in checkpoints/ dir, but I am not sure how to use this.

Thank you

phamquiluan commented 7 months ago

You can check this file to see if it helps: https://github.com/phamquiluan/ResidualMaskingNetwork/blob/master/legacy/demo_one_image.py

Daniel-CG-Wright commented 7 months ago

Thank you for the fast response, this helps!