PB模型转换RKNN 问题

[复制链接] · 发表于 2021-7-6 15:43:49

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脚本说不对

只看该作者 · 发表于 2021-11-11 14:12:14

netron就可以