|
在Ubuntu环境下将TensorFlow1.11.0版本生成的yolov3 pb文件转换为rknn文件,过程中如果不开启预编译,rknn模型在开发板上能够正常推理,结果正确。但是一旦开启预编译,转换得到的rknn模型在3399pro开发板上加载后无法推理。经过排查,运行过程卡在模型推理rknn.inference(inputs=[img])这步。
代码附上,其中supermarket.py为开发板上的运行代码。supermarket.rknn为已量化未开启预编译模型(能够推理识别,结果正确,附测试图片)。pre_compile.rknn(开启预编译,运行时卡住无法正常推理识别)。pre_compile.py为模型转换代码,yolov3_test_loss=1.0708.pb为原模型文件
链接:https://pan.baidu.com/s/1_jHW4l1_n68qn7_fCz-baQ
提取码:8888
|
|