DDRNet.TensorRT icon indicating copy to clipboard operation
DDRNet.TensorRT copied to clipboard

关于上采样的问题

Open Jas-zheng opened this issue 4 years ago • 2 comments

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

Jas-zheng avatar May 25 '21 02:05 Jas-zheng

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

  1. 可以在原作者开源的训练模型(双线性差值)基础上,微调几个epoch应该会有一定提升,提升肯定不会很大,这里用的是双线性差值的模型转tensorrt,tensorRT的上采样直接采用的nearest,效果差别其实不大.像yolov5的pytorch原始的上采样也是用的nearest,所以感觉没必要追求双线性差值的提升;
  2. 可以插件实现双线性差值,网上应该有很多类似的源码可以直接拿来用.

midasklr avatar May 31 '21 06:05 midasklr

好的,感谢大神的解答与建议,我会进行尝试的

Jas-zheng avatar May 31 '21 11:05 Jas-zheng