Transfer Learning colab notebook for mobilenetv1

I have created a Colab notebook for transfer learning using the flower dataset using mobilenet v1. The end result is a kmodel file which you can upload to Maixpy. Enjoy. Any improvement and comment is welcome.

Transfer Learning

3 Likes

nice work~

1 Like

Thank you very much for sharing such a good example! I want to share a slightly modified version of your code for M5StickV (the built-in LCD is 240 × 135 pixels). :wink:

1 Like

BTW, I’ve tested with the (experimental?) minimum firmware for M5StickV, instead of the official version introduced in the quick start.

Hi,

I just published a Google Colab notebook based on @gheesung 's notebook.

https://colab.research.google.com/drive/1mirG8BSoB3k87mh-qyY3-8-ZXj0XB6h6

I extended the original version with support for general data sets and visualization with Grad-CAM. Comments and suggestions are welcome. :wink:

1 Like

Thank you for the work.

@kotobuki By the way, I have modified the mobilenet V2. The model can be converted to kmodel but it hanged during inference. I am not sure what had happened, if you’re interested, I can share with you the notebook privately.

1 Like