|
- if __name__ == '__main__':
- # Create RKNN object
- rknn = RKNN()
- ret = rknn.load_rknn(path='./mobilenet_v1.rknn')
- # Set inputs
- img = cv2.imread('./dog_224x224.jpg')
- img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
-
- # init runtime environment
- print('--> Init runtime environment')
- ret = rknn.init_runtime(target='rk3399pro')
- if ret != 0:
- print('Init runtime environment failed')
- exit(ret)
- print('done')
-
- # Inference
- print('--> Running model')
- outputs = rknn.inference(inputs=[img])
- show_outputs(outputs)
- print('done')
-
- # perf
- print('--> Begin evaluate model performance')
- perf_results = rknn.eval_perf(inputs=[img])
- print('done')
-
- rknn.release()
如上图一段代码,在1808上运行正常。如今想在3399pro上运行,其中ret = rknn.init_runtime(target='rk1808')已经改成rk3399pro
3399开发版的运行环境是默认的fedora28, rknn已安装,使用python3运行该段代码时报错,截图如下:
请问这种问题如何解决? adb和ntb两种模式有什么区别? npu_transefer_proxy只能适用于其中一种?
|
本帖子中包含更多资源
您需要 登录 才可以下载或查看,没有帐号?立即注册
x
|