search

midasklr / DDRNet.TensorRT

TensorRT of DDRNet for real-time segmentation
38 stars 11 forks source link

关于上采样的问题 #1

Open Jas-zheng opened 3 years ago

Jas-zheng commented 3 years ago

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

midasklr commented 3 years ago

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

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

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