tensorRT_Pro 大佬你好，如果我想在yolov5中多加一层检测层，这时候我该怎么去修改你的代码呢？

大佬你好，如果我想在yolov5中多加一层检测层，这时候我该怎么去修改你的代码呢？

Open Anoue opened this issue 2 years ago • 4 comments

Nov 25 '22 08:11 Anoue

不需要修改啊，正常导出使用

Nov 28 '22 12:11 shouxieai

我增加了一个检测层后，好像并没有成功生成trtmodel文件，后续如果我想增加注意力机制什么的，我需要修改什么文件呀？

Nov 29 '22 02:11 Anoue

你好，请问你解决问题了吗，我也自己修改了网络，但没法从.onnx转到trt模型

Dec 05 '22 16:12 myAstar

不需要修改啊，正常导出使用

请问自己修改后的网络也可以直接导出吗，改了一些卷积模块

Dec 05 '22 16:12 myAstar