柚子快報邀請碼778899分享：TinyML之micro

http://yzkb.51969.com/

具體的開發(fā)過程網(wǎng)上都有，就不贅述了。

在colab上訓(xùn)練模型，花了將近2小時，得到模型文件后，CV到micro_speech工程里的micro_features_model.cpp里，結(jié)果運(yùn)行報錯：

Feature generation failed

Requested feature_data_ size 536907080 doesn't match 1960

最終在github上找到了解決方法，記錄一下，在micro_speech.ino中作如下修改：

static tflite::MicroMutableOpResolver<5> micro_op_resolver(error_reporter);

if (micro_op_resolver.AddDepthwiseConv2D() != kTfLiteOk) {

return;

}

if (micro_op_resolver.AddFullyConnected() != kTfLiteOk) {

return;

}

if (micro_op_resolver.AddSoftmax() != kTfLiteOk) {

return;

}

if (micro_op_resolver.AddReshape() != kTfLiteOk) {

return;

}

if (micro_op_resolver.AddConv2D() != kTfLiteOk) {

return;

}

其實(shí)就是增加了

if (micro_op_resolver.AddConv2D() != kTfLiteOk) { return; }

并把

static tflite::MicroMutableOpResolver<4> micro_op_resolver(error_reporter);

改成

static tflite::MicroMutableOpResolver<5> micro_op_resolver(error_reporter);

柚子快報邀請碼778899分享：TinyML之micro

http://yzkb.51969.com/

相關(guān)文章