Toybrick

PB模型转换RKNN 问题

wenzhu

注册会员

积分
58
发表于 2021-7-6 15:43:49    查看: 6320|回复: 1 | [复制链接]    打印 | 显示全部楼层
PB 模型转换成RKNN模型时,需要知道INPUTS 和OUTPUTS,想问下这两个值可以从哪里查看得知?
本来以为通过命令:bazel-bin/tensorflow/tools/graph_transforms/summarize_graph   --in_graph=xxx.pb
Found 1 possible inputs: (name=image_tensor, type=uint8(4), shape=[?,?,?,3])
No variables spotted.
Found 4 possible outputs: (name=detection_boxes, op=Identity) (name=detection_scores, op=Identity) (name=detection_classes, op=Identity) (name=num_detections, op=Identity)
得出的inputs和outputs就是,可惜放进转换的Python脚本说不对


回复

使用道具 举报

kepurSong

中级会员

积分
423
发表于 2021-11-11 14:12:14 | 显示全部楼层
netron就可以
回复

使用道具 举报

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

本版积分规则

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


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