|
|
【人工智能】
pb文件导出rknn文件出错
发表于 2021-5-10 11:48:32
浏览：7285
|
回复：1
打印
只看该作者
[复制链接]
楼主
tensorflow训练了一个全连接神经网络,参数文件是pb,在板子上能够正常转成rknn文件。
后来为了用卷积网络，对输入的灰度图片数据增添了一个维度，【None,28,28】-->【None,28,28,1】.网络训练完成在转rknn时，我设置rknn.load_tensorflow中的input_size_list=[[28,28,1]].报了如下错误
--> Building model
W The target_platform is not set in config, using default target platform rk1808.
I Generate input meta
I Load input meta
W The channel_mean_value filed will not be used in the future!
E The channel_mean_value length of input 1 must equal 2, such as "0 1" (MEAN SCALE)!
--> Export RKNN model
E RKNN model data is None, please load model first!
E RKNN model data is None, please load model first!
Export rknn failed! |
|
