maix_train
maix_train copied to clipboard
The solution of model breakpoint continuous training(断点续训,无需重新训练)
Take mobilenet's classification model as an example
Remember to backup!
- By default, after your training, there is a
m.tflite.h5
file inmaix_train/out
. Copy it to some place else. - You need to modify lines 121 to 129 in
maix_train/train/classifier/__init__.py
- Load the model with
self.model = tf.keras.models.load_model("Your filepath of *.h5 ")
Mobilenet的分类模型测试过可行
记得备份!
- 默认情况下,在你训练之后,在
maix_train/out
中有一个m.tflite.h5
文件。 将其复制到其他地方。 - 需要修改
maix_train/train/classifier/__init__.py
中的121到129行 - 使用
self.model = tf.keras.models.load_model("Your filepath of *.h5")
加载模型