search

sipeed / maix_train

k210(MaixPy)/V831 model example train code, include mobilenet classifier and YOLO V2 detector
https://wiki.sipeed.com/maixpy
Apache License 2.0
83 stars 40 forks source link

The solution of model breakpoint continuous training（断点续训，无需重新训练） #21

Open kissablemt opened 2 years ago

kissablemt commented 2 years ago

Take mobilenet's classification model as an example

Remember to backup!

  1. By default, after your training, there is a m.tflite.h5 file in maix_train/out. Copy it to some place else.
  2. You need to modify lines 121 to 129 in maix_train/train/classifier/__init__.py
  3. Load the model with self.model = tf.keras.models.load_model("Your filepath of *.h5 ")

image

Mobilenet的分类模型测试过可行

记得备份！

  1. 默认情况下，在你训练之后，在 maix_train/out 中有一个 m.tflite.h5 文件。 将其复制到其他地方。
  2. 需要修改maix_train/train/classifier/__init__.py中的121到129行
  3. 使用self.model = tf.keras.models.load_model("Your filepath of *.h5")加载模型