|
请问一下,rknn-toolkit 跟 tensorflow 的版本是否必须一一对应,就是我使用某个固定版本的rknn-toolkit 搭建两个不同版本tensorflow的环境,对于相同的op,在这两个不同版本的tensorflow下是否都可以正常转换?本人遇到的具体问题是:
1.使用docker里面的环境,tensorflow是1.11版本,可以正常load 用1.11生成的tflite文件,同时能成功export_rknn;(tflite不管是否做量化都可以成功load及export, rknn的接口统一不做量化);
2.但是如果用conda自己搭建环境,rknn是同一个版本,tensorflow换成高一点的版本,r.1.14, 这时候可以load用1.14生成的tflite,但是build rknn失败,错误如下:
T Traceback (most recent call last):
T File "rknn/api/rknn_base.py", line 515, in rknn.api.rknn_base.RKNNBase.build
T File "rknn/api/rknn_base.py", line 439, in rknn.api.rknn_base.RKNNBase._build
T File "rknn/base/ovxconfiggenerator.py", line 173, in rknn.base.ovxconfiggenerator.generate_vx_config_from_files
T File "rknn/base/RKNNlib/RKNNnet.py", line 330, in rknn.base.RKNNlib.RKNNnet.RKNNNet.norm_parameters
T File "rknn/base/RKNNlib/layer/convolution.py", line 36, in rknn.base.RKNNlib.layer.convolution.Convolution._norm_params
T AttributeError: 'NoneType' object has no attribute 'format'
Build rknn failed!
是否因为tf版本问题,rknn转换对应的关键字不一致等原因,导致不能转换?
其他参考信息:
rknn-toolkit 1.0
转换的模型算法是inception v3,没有特殊op(这里用load_tensorflow的接口转换高版本和低版本的pb文件都没问题的, 目前是不支持转换高版本的tflite文件,低版本的tflite可以)
|
|