测试IMX6UL上跑Tensorflow Lite (跑分)

/ 1评 / 1

最近发现一个配方,给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

  1. zwmasdf说道:

    可以用c跑tflite,速度会比python快很多

回复 zwmasdf 取消回复

您的电子邮箱地址不会被公开。 必填项已用 * 标注