最近发现一个配方,给Yotco构建系统添加一个Tensorflow Lite,还支持添加Python API和C API,默认是为了匹配树莓派的,匹配其他也是相当的方便,因此,我就简单Port了一下.
源工程:https://github.com/NobuoTsukamoto/meta-tensorflow-lite
然后在配置中新增包引入,具体就自由取舍了.
FORTRAN:forcevariable = ",fortran"
IMAGE_INSTALL:append = "python3-tensorflow-lite \
libtensorflow-lite \
tensorflow-lite-benchmark \
tensorflow-lite-minimal \
tensorflow-lite-label-image \
python3-tensorflow-lite-example \
"
因为是单核的,只跑一个线程试试看.
也跑了一下图像标记,大概3秒能标记一个,Mobilenet-224,大家觉得怎样呢?当然不能和NPU比了.
也跑了跑Python版本的,结果似乎就没那么好看了.(应该是CPU弱鸡导致Python到底层之间传递开销太大了!)
我的IMX6UL工程:https://github.com/nickfox-taterli/imx-yocto-bsp
可以用c跑tflite,速度会比python快很多