search

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

自己训练的模型检测框错位，标签乱 #107

Open PhenixFeather opened 3 months ago

PhenixFeather commented 3 months ago

我基于ultralytics的yolov8代码修改了一下yolov8的结构，并在自己的数据集上进行训练得到模型，类别与原模型有出入。

用官方的脚本导出为onnx，然后通过trtexec导出为trt模型后，检测单张图片时，出现检测框错位与标签乱码的现象，视频推理时，cuda核心报错。

请问这个问题，我是需要修改代码的什么地方吗？

abcc235 commented 3 months ago

我也遇到了这个问题