tensorflow-yolov3 自己训练衣服类型检测，可以生成ckpt,，但是转pb有问题，output_node

自己训练衣服类型检测，可以生成ckpt,，但是转pb有问题，output_node_names是什么？

Open hitxuhao opened this issue 4 years ago • 3 comments

提供的convert_weight.py，显示"input/input_data", "pred_sbbox/concat_2", "pred_mbbox/concat_2", "pred_lbbox/concat_2不在图中

Jun 09 '20 12:06 hitxuhao

output_node_names = ["input/input_data", "conv_sbbox/Conv2D", "conv_mbbox/Conv2D", "conv_lbbox/Conv2D"]

Jun 12 '20 07:06 robinjoe93

@robinjoe93 When I use those output node names, I get error saying they don't exist. What are the correct names?

Dec 22 '20 18:12 ajaymaity

对的，训练自己数据集的时候，在ckpt转pb（无论是convert weights或者freeze_graph）都是有问题的

Mar 19 '21 06:03 laisimiao