search

shouxieai / tensorRT_Pro

C++ library based on tensorrt integration
MIT License
2.62k stars 549 forks source link

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

Open Anoue opened 2 years ago

shouxieai commented 2 years ago

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

Anoue commented 2 years ago

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

myAstar commented 1 year ago

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

myAstar commented 1 year ago

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

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