TensorFlow Lite for Microcontrollers
使用方法
参考tensorflow/lite/micro/examples/xxx 目录下的使用方法, 以hello_world为例,
文件hello_world_test.cc
1. 创建MicroErrorReporter object
tflite::MicroErrorReporter micro_error_reporter;
2. 有tflite model文件得到 tflite::Modle 结构体
const tflite::Model* model = ::tflite::GetModel(g_model);
3. 创建OpsResolver对象
这里是创建包含所有算子的对象
tflite::AllOpsResolver resolver;
如果是根据实际使用的算子创建OpsResolver, 使用
static tflite::MicroMutableOpResolver<5> micro_op_resolver; micro_op_resolver.AddConv2D();
共有 0 条评论