search

av-savchenko / hsemotion

HSEmotion Python library for facial expression recognition
Apache License 2.0
23 stars 7 forks source link

MobileNet model is supported #3

Open rageSpin opened 1 year ago

rageSpin commented 1 year ago

Hello everyone, thanks for this great work!

In your face emotion recognition repository, there is also MobileNet model weights. However, in this repository it is not supported as model_name. This is a pity since is the most efficient model if we consider the inference time vs. accuracy.

Can it be added in some way?

Thanks

av-savchenko commented 1 year ago

Hello! Thanks for your interest! MobileNet is very different from other models because MobileNet was trained in the TensorFlow, while EfficientNet models were trained in PyTorch. I decided not to add Tensorflow's dependencies into hsemotion library, and I'm not sure that I will have time to train MobileNet in PyTorch soon. The only one workaround now is to use the MobileNet model in your code similarly to display_emotions.ipynb