Firefly开源社区

[人工智能] pb文件导出rknn文件出错

6

积分

0

威望

0

贡献

技术小白

积分
6
发表于 2021-5-10 11:48:32     
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!
回复

使用道具 举报

6

积分

0

威望

0

贡献

技术小白

积分
6
发表于 2021-5-10 15:40:11     
我把维度扩展到3,能转了
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

友情链接 : 爱板网 电子发烧友论坛 云汉电子社区 粤ICP备14022046号-2
快速回复 返回顶部 返回列表