如何将这个推理框架应用到我自己训练的yolov7模型上，需要进行什么更改呢？

FeiYull / TensorRT-Alpha

🔥🔥🔥TensorRT for YOLOv8、YOLOv8-Pose、YOLOv8-Seg、YOLOv8-Cls、YOLOv7、YOLOv6、YOLOv5、YOLONAS......🚀🚀🚀CUDA IS ALL YOU NEED.🍎🍎🍎

GNU General Public License v2.0

1.25k stars 195 forks source link

Closed huanglong161 closed 4 months ago

huanglong161 commented 4 months ago

比如我训练后的模型尺寸是384*640，类别是两类，麻烦您告诉我该怎么样进行更改呢？

huanglong161 commented 4 months ago

我将外部的num数量进行了更改，将size也改成了384和640，模型推理过程中没有报错，但是推理的结果是错误的，请问还需要在哪里进行更改吗

huanglong161 commented 4 months ago

问题出现在trt上，需要重新生成一个trt，推理就正确了