Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

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

Open
PhenixFeather opened this issue Apr 5, 2024 · 2 comments
Open

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

PhenixFeather opened this issue Apr 5, 2024 · 2 comments

Comments

@PhenixFeather
Copy link

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

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

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

abcc235 commented Apr 10, 2024

我也遇到了这个问题

@ziqiang-Cao
Copy link

Is there a solution?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
3 participants
@PhenixFeather @abcc235 @ziqiang-Cao