zz511 发表于 2021-5-10 11:48:32

pb文件导出rknn文件出错

tensorflow训练了一个全连接神经网络,参数文件是pb,在板子上能够正常转成rknn文件。
后来为了用卷积网络,对输入的灰度图片数据增添了一个维度,【None,28,28】-->【None,28,28,1】.网络训练完成在转rknn时,我设置rknn.load_tensorflow中的input_size_list=[].报了如下错误
--> 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!

zz511 发表于 2021-5-10 15:40:11

我把维度扩展到3,能转了
页: [1]
查看完整版本: pb文件导出rknn文件出错