Firefly开源社区

标题: pb文件导出rknn文件出错 [打印本页]

作者: zz511    时间: 2021-5-10 11:48
标题: pb文件导出rknn文件出错
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!
作者: zz511    时间: 2021-5-10 15:40
我把维度扩展到3,能转了




欢迎光临 Firefly开源社区 (https://dev.t-firefly.com/) Powered by Discuz! X3.1