search

google-coral / tutorials

Colab/Jupyter tutorials about training TensorFlow models for Edge TPU, and other tutorials
Apache License 2.0
181 stars 39 forks source link

Mobilenet v3 model tutorial #15

Closed omerferhatt closed 3 years ago

omerferhatt commented 3 years ago

Can someone add Mobilenet v3 model implementation on TF2 because Keras pre-trained model not working.

omerferhatt commented 3 years ago

Hello again, the main problem is model.input.set_shape((1,) + model.input.shape[1:])

In MobileNetV2, constructed model can be compiled without editing. But in V3Small and V3Large, model input shape needs to be set before saving model. After setting batch shape, model compiled successfully.

hjonnala commented 3 years ago

Hi, It seems you found the solution. Are you still looking for Mobilenet v3 model implementation on TF2?

omerferhatt commented 3 years ago

Hello, actually no. It's working now. But it would be great if docs will be updated for newcomers.

hjonnala commented 3 years ago

Sure, we will make the changes when tutorial got updated with moblenetv3.

For others, for any edgetpu compiler conversion issues please refer to this tutorial.

Thanks