tensorflow-yolov3 tensorRT不支持tensorflow的内置函数

tensorRT不支持tensorflow的内置函数

Open hongtao-ai opened this issue 5 years ago • 6 comments

用tensorRT进行模型加速，但是tensorRT不支持tensorflow的tf.cast，tf.tile，tf.range等函数，这几个函数在yolov3.py里，不知道用什么方法改成tensorRT能识别的函数

Nov 26 '19 14:11 hongtao-ai

从前面的节点分段，后面的处理自己实现

Dec 03 '19 12:12 XuVision

用tensorRT进行模型加速，但是tensorRT不支持tensorflow的tf.cast，tf.tile，tf.range等函数，这几个函数在yolov3.py里，不知道用什么方法改成tensorRT能识别的函数

请问你解决了吗？

Apr 28 '20 03:04 lergliu

做的不是tensorRT的，但是处理后结果是正确的

Apr 28 '20 03:04 XuVision

做的不是tensorRT的，但是处理后结果是正确的

怎么做的，能否指导下？

Apr 29 '20 00:04 lergliu

网络裁剪到后面的13×13×255、26×26×255、52×52×255 3个feature map，然后根据算法原理计算框的大小最后做nms

Apr 29 '20 01:04 XuVision

請問您是如何用tensorRT进行模型加速,可以分享嗎

Feb 07 '23 03:02 s0966066980