How to save the classification head?

leondgarse / Keras_insightface

Insightface Keras implementation

MIT License

230 stars 56 forks source link

How to save the classification head? #114

Closed RuoyuChen10 closed 1 year ago

RuoyuChen10 commented 1 year ago

Hi, thanks for your contribution. I want to know how to save the classification head used for identification, what should I operate? Thanks!

leondgarse commented 1 year ago

After training, there will be 2 models being saved, basic_model and model, and the classifier header for training is saved with model, as described in basic-training Model part. The model is saved as ./checkpoints/{save_path.h5} and basic_model ./checkpoints/{save_path}_basic_agedb30_xxx_epoch_xxx.h5.
For running eval or real world recognition, the header is not exists, refer video_test.py#L90-L105 for usage.

RuoyuChen10 commented 1 year ago

After training, there will be 2 models being saved, basic_model and model, and the classifier header for training is saved with model, as described in basic-training Model part. The model is saved as ./checkpoints/{save_path.h5} and basic_model ./checkpoints/{save_path}_basic_agedb30_xxx_epoch_xxx.h5.

For running eval or real world recognition, the header is not exists, refer video_test.py#L90-L105 for usage.

Thanks, my problem has been well addressed.