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/)