MobileNetV2 about mobilenet_v1_stmcube_ai HOT 5 OPEN

Ciao @EduardShaid,

This project starts from a pre-trained Mobilenet model downloaded by the module keras.applications. You can find the script here: https://github.com/alessandrocapotondi/mobilenet_stmcube_ai/blob/master/keras_model/keras_mobilenet_stm_cubemx_ai.ipynb

The main problem is that not all the layers are supported by STMCubeMX.AI. For example it do not support ReLU directly. Such layers require to be converted to Activation('relu').
This operation is done as well by the script above.

By the way, at the moment I am not able to link the generated code because contrary to what STMCubeMX.AI the activations do not inside the SRAM...I am posting an issue right now on the STM Community Forum.

About the MobileNetV2, I did test it, so I do not have so much feedbacks. Sorry.

Alessandro

from mobilenet_v1_stmcube_ai.

The launch of the test_and_compare_models(...) function shows that the old model and the new one work differently ! It turns out that after the transformation of the model, it stops working correctly. Is it so ? and is it necessary to retrain the model ?

from mobilenet_v1_stmcube_ai.

Ciao @EduardShaid,

You don't need to retrain the model. There was an issue on the conversion. I was using unbounded relu activations while the pretrained model was based on relu6.

I fixed the model and the script on the last commit 1adeb45.

Thanks,
Alessandro

from mobilenet_v1_stmcube_ai.

Yes, thank you, the example works.
I have another question about MobilenetV2: can I use the already trained MobilenetV2 model, but converted by your method, to recognize faces on the STM32 platform ?

from mobilenet_v1_stmcube_ai.

Can you give an example of your approach for MobileNetV2 ?
It does not work out for me.

from mobilenet_v1_stmcube_ai.