菜橙 2022-08-04 09:53 采纳率: 50%
浏览 63
已结题

修改了backbone的yolov5如何进行tensorRT部署？

修改了backbone的yolov5如何进行tensorRT部署？原有的方法在模型修改结构后不支持了。

  • 写回答

1条回答 默认 最新

  • kakaccys 2022-08-05 19:33
    关注

    tensorrt默认支持一些常规层操作，如果需要支持修改的层，需要写自定义层，再注册进去，可以自行搜索tensorrt添加自定义层方法，有很多教程

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论

报告相同问题？

问题事件

  • 系统已结题 8月19日
  • 已采纳回答 8月11日
  • 创建了问题 8月4日