Toybrick

.pb 转为 .rknn 报错

xumiaoyu

新手上路

积分
46
发表于 2019-10-14 16:54:04    查看: 15791|回复: 11 | [复制链接]    打印 | 显示全部楼层

本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有帐号?立即注册

x
回复

使用道具 举报

jefferyzhang

版主

积分
12850
发表于 2019-10-14 21:39:00 | 显示全部楼层
看log是你文件路径写错了吧
回复

使用道具 举报

xumiaoyu

新手上路

积分
46
 楼主| 发表于 2019-10-15 09:28:42 | 显示全部楼层
jefferyzhang 发表于 2019-10-14 21:39
看log是你文件路径写错了吧

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

使用道具 举报

xumiaoyu

新手上路

积分
46
 楼主| 发表于 2019-10-15 09:32:46 | 显示全部楼层
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

版主

积分
12850
发表于 2019-10-15 10:45:10 | 显示全部楼层
1. 模型是哪里来的,是自己保存的么,还是其他github里的预训练模型,还是从其他格式转过来的?
2. 这个pb模型试过用tf载入能用么?
回复

使用道具 举报

xumiaoyu

新手上路

积分
46
 楼主| 发表于 2019-10-15 15:14:33 | 显示全部楼层
本帖最后由 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.

本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有帐号?立即注册

x
回复

使用道具 举报

xumiaoyu

新手上路

积分
46
 楼主| 发表于 2019-10-15 15:20:50 | 显示全部楼层
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

版主

积分
12850
发表于 2019-10-15 15:36:46 | 显示全部楼层
xumiaoyu 发表于 2019-10-15 15:14
.ckpt模型在github上下载的。把.ckpt 转.pb. 日志文件中有ValueError:I/O operation on closed file. ...

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

使用道具 举报

xumiaoyu

新手上路

积分
46
 楼主| 发表于 2019-10-16 10:22:37 | 显示全部楼层
jefferyzhang 发表于 2019-10-15 15:36
这问题你要自己解决了,只能说明你转的pb就已经有问题了,在转rknn那是肯定有问题的 ...

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

使用道具 举报

xumiaoyu

新手上路

积分
46
 楼主| 发表于 2019-10-16 10:36:27 | 显示全部楼层
xumiaoyu 发表于 2019-10-16 10:22
文档例子中的.pb文件是怎么来的,是训练过程中直接生成的,还是训练时生成.ckpt再转为.pb? ...

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

使用道具 举报

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

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表