TensorRT-Alpha
TensorRT-Alpha copied to clipboard
自己训练的模型检测框错位,标签乱
我基于ultralytics的yolov8代码修改了一下yolov8的结构,并在自己的数据集上进行训练得到模型,类别与原模型有出入。
用官方的脚本导出为onnx,然后通过trtexec导出为trt模型后,检测单张图片时,出现检测框错位与标签乱码的现象,视频推理时,cuda核心报错。
请问这个问题,我是需要修改代码的什么地方吗?