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.

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

Update for yolov10 support #23

Open
AekSW opened this issue May 30, 2024 · 1 comment
Open

Update for yolov10 support #23

AekSW opened this issue May 30, 2024 · 1 comment

Comments

@AekSW
Copy link

AekSW commented May 30, 2024

Is it possible to use yolov10? I tried using it but it broke the utils for Bytetrack.

Traceback (most recent call last):
  File "C:\Projects\Vehicle-CV-ADAS-master\demo.py", line 304, in <module>
    objectTracker.DrawTrackedOnFrame(frame_show, False)
  File "C:\Projects\Vehicle-CV-ADAS-master\ObjectTracker\byteTrack\byteTracker.py", line 213, in DrawTrackedOnFrame
    self.plot_trajectories(frame, trajector , cid, tid)
  File "C:\Projects\Vehicle-CV-ADAS-master\ObjectTracker\core.py", line 195, in plot_trajectories
    color=self.class_colors[class_id],
KeyError: 'unknown'

What can I do to fix it? Will there be any update for yolov10 support?

@jason-li-831202
Copy link
Owner

YOLOv10 has added some post-processing parts compared to the previous YOLO series. Therefore, there are two ways to use it: either delete the original model's post-processing without converting it to ONNX, or write new post-processing for YOLOv10's output.

image

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants