tensorflow-yolov3 yolov3转为为pb模型的时候如何获取尺寸信息？

yolov3转为为pb模型的时候如何获取尺寸信息？

Open float123 opened this issue 5 years ago • 7 comments

我尝试用tensorrt对yolov3进行加速，利用tensorflow自带的api将yolov3_coco.pb转为为yolov3_trt.pb，但是出现一个问题，程序无法识别模型的尺寸，会出现[？,?,?,3,2]这种情况。

然后我查了一下网络结构（最下图是能转化的模型），发现缺少一些尺寸信息，请问要如何获取这些type信息呢？感谢大佬了

Jul 15 '19 08:07 float123

你用的TensorRT几？我之前也做过TensorRT加速这块，可以加微信交流 yy1186584388

Jul 15 '19 15:07 YunYang1994

@float123 大佬用的是什么visualization tool啊？感谢！

Jul 27 '19 00:07 yuanzhedong

@ float123大佬用的是什么可视化工具啊？感谢！

tensorboard and Netron

Jul 29 '19 02:07 float123

请问自己训练的模型参数怎么转换成PB文件?谢谢

Dec 06 '19 08:12 scuizhibin

你好我将模型转化为Pb文件后，尺度也无法识别。请问你解决了这个问题吗？谢谢

Jul 29 '20 05:07 zhouxianming

@zhouxianming

Using these

$ python convert_weight.py $ python freeze_graph.py

Sep 17 '20 03:09 MuhammadAsadJaved

@float123 Can you tell me please how we can convert the yolov3 TensorFlow model to TRT for acceleration?

Sep 17 '20 03:09 MuhammadAsadJaved