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

Correct naming of YOLOv8 output layers #33

Open
maxsitt opened this issue Feb 28, 2023 · 1 comment
Open

Correct naming of YOLOv8 output layers #33

maxsitt opened this issue Feb 28, 2023 · 1 comment
Labels
enhancement New feature or request

Comments

@maxsitt
Copy link

maxsitt commented Feb 28, 2023

Output layers of the YOLOv8 ONNX model are currently named output1_yolov6r2, output2_yolov6r2, output3_yolov6r2.
https://github.com/luxonis/tools/blob/master/yolo/export_yolov8.py#L60

Not sure if this is on purpose for inference to work on the OAK. Correct naming of the output layers (output1_yolov8 etc.) might make things clearer for manual conversion afterwards.

@tersekmatija
Copy link
Collaborator

Hey, good catch! This is indeed intentional. Since both use the same processing, we accelerated the support for YoloV8 by naming them like YoloV6 layers. This will be fixed once support on FW (in our DepthAI) for YoloV8 directly is added with the new release.

@tersekmatija tersekmatija added the enhancement New feature or request label Feb 28, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

2 participants