Toybrick

标题: .pb 转为 .rknn 报错 [打印本页]

作者: xumiaoyu 时间: 2019-10-14 16:54
标题: .pb 转为 .rknn 报错
[attach]641[/attach]

作者: jefferyzhang 时间: 2019-10-14 21:39
看log是你文件路径写错了吧

作者: xumiaoyu 时间: 2019-10-15 09:28

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

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

作者: xumiaoyu 时间: 2019-10-15 09:32
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')

作者: jefferyzhang 时间: 2019-10-15 10:45
1. 模型是哪里来的，是自己保存的么，还是其他github里的预训练模型，还是从其他格式转过来的？
2. 这个pb模型试过用tf载入能用么？

作者: xumiaoyu 时间: 2019-10-15 15:14
本帖最后由 xumiaoyu 于 2019-10-15 15:21 编辑

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:20

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

作者: jefferyzhang 时间: 2019-10-15 15:36

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

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

作者: xumiaoyu 时间: 2019-10-16 10:22

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

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

作者: xumiaoyu 时间: 2019-10-16 10:36

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

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

作者: jefferyzhang 时间: 2019-10-17 08:57

xumiaoyu 发表于 2019-10-16 10:36
我是在github下的工程，在网上找到代码把.ckpt 转为 .pb 的。

所以你试过这个pb文件可以正常被tf载入推理么？

作者: raul 时间: 2019-10-18 09:15

xumiaoyu 发表于 2019-10-16 10:36
我是在github下的工程，在网上找到代码把.ckpt 转为 .pb 的。

你是怎么转的，有做freeze吗

欢迎光临 Toybrick (https://t.rock-chips.com/)