DDRNet.TensorRT

DDRNet.TensorRT copied to clipboard

为什么使用wts文件

1

如果使用ddrnet.pytorch里面，转成onnx, 再用trtexec 转换成engine文件，推理就可以吧？

northeastsquare

想提升输入-输出的总体时间

JJY0710


mx1mx2

Hi! The getwts.py script is unusable due to the required modules are not included in the project. Could you upload them or just the DDRNet_CS.wts file to google drive? Thanks!

sync24122

int8 跑出来结果全黑

1

是不是模型对int8 不支持，如果想跑出int8的速度 需要改训练时的模型，使其最后得到的参数已经是量化的。 然后这个.tensorRT的作用。。。就 是不是和 ddrnet 直接导出 .onnx,然后用官方自带的工具转成trt 效果是一样的

ghost

Could you give a google drive link for the .wts file? thanks

3

I cannot use baidunetdisk binaries...

Tetsujinfr

DDRNet Support DDRNet23?

2

I was wondering if this could support ddrnet23 (not slim) as well

hj1295

关于上采样的问题

2

感谢大神提供的代码，受益匪浅！ 关于上采样我有个疑问，想恢复准确率的话需要finetune with nearest是指将训练模型的上采样方式改成nearest吗？然后有没有可能在tensorrt的c++代码内修改上采样方式为bilinear呢

Jas-zheng