.pb 转为 .rknn 报错

看log是你文件路径写错了吧

jefferyzhang 发表于 2019-10-14 21:39
看log是你文件路径写错了吧

路径没有错啊，完全按照例子改写的。

rknn = RKNN(verbose=True)
rknn.config(channel_mean_value='128 128 128 128', reorder_channel='0 1 2')
print('--> Loading model')
rknn.load_tensorflow(tf_pb='./model.pb',
                         inputs=['resnet_v1_101/Pad'],
                         outputs=['resnet_v1_105_5/bbox_pred/BiasAdd'],
                         input_size_list=[[INPUT_SIZE, INPUT_SIZE, 3]]
print('done')
# Build Model
print('--> Building model')
rknn.build(do_quantization=True, dataset='./dataset.txt')
print('done')
# Export RKNN Model
rknn.export_rknn('./ssd_mobilenet_face.rknn')

1. 模型是哪里来的，是自己保存的么，还是其他github里的预训练模型，还是从其他格式转过来的？
2. 这个pb模型试过用tf载入能用么？

jefferyzhang 发表于 2019-10-15 10:45
1. 模型是哪里来的，是自己保存的么，还是其他github里的预训练模型，还是从其他格式转过来的？
2. 这个pb ...

.ckpt模型在github上下载的。把.ckpt 转.pb. 日志文件中有ValueError:I/O operation on closed file.

xumiaoyu 发表于 2019-10-15 15:14
.ckpt模型在github上下载的。把.ckpt 转.pb. 日志文件中有ValueError:I/O operation on closed file. ...

H:/2019-10-15 13-59-20 的屏幕截图.png

xumiaoyu 发表于 2019-10-15 15:14
.ckpt模型在github上下载的。把.ckpt 转.pb. 日志文件中有ValueError:I/O operation on closed file. ...

这问题你要自己解决了，只能说明你转的pb就已经有问题了，在转rknn那是肯定有问题的

jefferyzhang 发表于 2019-10-15 15:36
这问题你要自己解决了，只能说明你转的pb就已经有问题了，在转rknn那是肯定有问题的 ...

文档例子中的.pb文件是怎么来的，是训练过程中直接生成的，还是训练时生成.ckpt再转为.pb？

xumiaoyu 发表于 2019-10-16 10:22
文档例子中的.pb文件是怎么来的，是训练过程中直接生成的，还是训练时生成.ckpt再转为.pb？ ...

我是在github下的工程，在网上找到代码把.ckpt 转为 .pb 的。