Rock 5b tensorflow lite NPU

Has anyone been able to run tensorflow lite on the NPU?

soemthing like this
https://www.tensorflow.org/lite/guide/python

1 Like

You need to convert the tensorflow lite model to rknn model using x86_64 version of rknn toolkit 2, then you can use the converted model.
Check this link: https://github.com/rockchip-linux/rknn-toolkit2

1 Like

this looks like a great start thanks. how do I see where the NPU is, like does it connect through PCI or soemthing?