TensorRT-Alpha icon indicating copy to clipboard operation
TensorRT-Alpha copied to clipboard

自己训练的模型检测框错位,标签乱

Open PhenixFeather opened this issue 10 months ago • 2 comments

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

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

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

PhenixFeather avatar Apr 05 '24 13:04 PhenixFeather