Toybrick

标题: PB模型转换RKNN 问题 [打印本页]

作者: wenzhu    时间: 2021-7-6 15:43
标题: PB模型转换RKNN 问题
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    时间: 2021-11-11 14:12
netron就可以




欢迎光临 Toybrick (https://t.rock-chips.com/) Powered by Discuz! X3.3