TFLM代码仓链接:https://github.com/tensorflow/tflite-micro
2.1 下载TFLM源代码
下载TFLM需要使用如下git命令:
git clone https://github.com/tensorflow/tflite-micro.git
TFLM顶层目录下的文件和目录,如下图所示:
2.2 TFLM基准测试说明
TFLM顶层目录有README.md文件,其Additional Documentation节列出来Benchmark说明,Benchmark(基准测试)用于衡量关键模型和工作负载的性能。
README.md前一半内容为:
完整内容参考: tflite-micro/tensorflow/lite/micro/benchmarks/README.md at main · tensorflow/tflite-micro (github.com)
2.3 TFLM基准测试命令
从README的”Run on x86”可以看到,在x86 PC上运行关键词基准测试的命令是:
make -f tensorflow/lite/micro/tools/make/Makefile run_keyword_benchmark
在x86 PC上运行人体检测基准测试的命令是:
make -f tensorflow/lite/micro/tools/make/Makefile run_person_detection_benchmark
以上两个命令都会调用make命令,并以tensorflow/lite/micro/tools/make/Makefile
为构建规则,分别构建run_keyword_benchmark
和run_person_detection_benchmark
两个目标。
查阅tensorflow/lite/micro/tools/make/Makefile
文件夹内容,可以看到:
这段代码中的 595行、601行、605行、612行、613行 分别会下载一些文件,具体下载的是tflm依赖的库和测试数据集;
而执行上面两个make命令,实际上会依次执行如下步骤:
- 下载依赖库和数据集;
- 编译测试程序;
- 运行测试程序;
必须至少一遍make命令,才会下载测试数据集,才能进行后续的移植步骤。