Deepstream_Project icon indicating copy to clipboard operation
Deepstream_Project copied to clipboard

大佬好,想问下更换了 backbone 并且剪枝过的 yolov5 适用于您这个项目吗

Open Veal98 opened this issue 3 years ago • 2 comments

如题

Veal98 avatar Nov 03 '21 07:11 Veal98

你的模型和Deepstream没关系,是Tensorrt你需要用tensorrt 去做转换,一个是ONNX到TRT的路线,一个是你把权重写进文件,然后自己写结构算子解析重构,这是tensortx有Demo参考。

positive666 avatar Dec 06 '21 08:12 positive666

想省事的话 用Yolov5的export去试试 可能需要调试修改下

positive666 avatar Dec 06 '21 08:12 positive666