search

Natsu-Akatsuki / RangeNetTrt8

tensorrt8 && cuda && libtorch implementation of rangenet++
MIT License
44 stars 9 forks source link

onnx转trt #12

Closed yuyaoliang closed 10 months ago

yuyaoliang commented 10 months ago

大佬您好，我看您在onnx转trt的时候有不同层的权值类型设置，这个应该是基于您自己的带有CRF的darknet53网络模型，我想问下如果我的darknet53网络模型不带CRF，应该怎么设置呢，我直接暴力的把这里注释掉后，效果会非常差，求指教 2023-10-25 11-21-27屏幕截图

Natsu-Akatsuki commented 10 months ago
  1. 这个模型不是我的，是原仓库的模型。
  2. 你注释了这里，也只是修改了对应层的权值策略，并不说你能去掉 CRF。
  3. 原仓库的模型，你怎么知道它是用了 CRF ？
yuyaoliang commented 10 months ago

感谢大佬，解决了，是我自己数据没有做相应修改，与转trt这里没关系