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

When will djl supports yolov7 and v8 #2740

Closed
andyliumathematics opened this issue Aug 9, 2023 · 2 comments
Closed

When will djl supports yolov7 and v8 #2740

andyliumathematics opened this issue Aug 9, 2023 · 2 comments
Labels
enhancement New feature or request

Comments

@andyliumathematics
Copy link

andyliumathematics commented Aug 9, 2023

Description

When will djl supports yolov7 and v8?
Now I test yolov7.pt translate onnx,it throws exception.
SLF4J: No SLF4J providers were found.
SLF4J: Defaulting to no-operation (NOP) logger implementation
SLF4J: See https://www.slf4j.org/codes.html#noProviders for further details.
SLF4J: Class path contains SLF4J bindings targeting slf4j-api versions 1.7.x or earlier.
SLF4J: Ignoring binding found at [jar:file:/Ccaches/modules-2/files-2.1/org.apache.logging.log4j/log4j-slf4j-impl/2.20.0/7ab4f082fd162f60afcaf2b8744a3d959feab3e8/log4j-slf4j-impl-2.20.0.jar!/org/slf4j/impl/StaticLoggerBinder.class]
SLF4J: See https://www.slf4j.org/codes.html#ignoredBindings for an explanation.
Exception in thread "main" ai.djl.translate.TranslateException: java.lang.ArrayIndexOutOfBoundsException: 5102
at ai.djl.inference.Predictor.batchPredict(Predictor.java:191)
at ai.djl.inference.Predictor.predict(Predictor.java:128)
at coOnnx.main(Onnx.java:54)
Caused by: java.lang.ArrayIndexOutOfBoundsException: 5102
at ai.djl.modality.cv.translator.YoloTranslator.processOutput(YoloTranslator.java:75)
at ai.djl.modality.cv.translator.YoloTranslator.processOutput(YoloTranslator.java:28)
at ai.djl.inference.Predictor.processOutputs(Predictor.java:306)
at ai.djl.inference.Predictor.batchPredict(Predictor.java:185)
... 2 more

@andyliumathematics andyliumathematics added the enhancement New feature or request label Aug 9, 2023
@KexinFeng
Copy link
Contributor

It looks feasible to reuse the yolov5 translator with small change. See #2753. If it works, you can consider contributing a PR to DJL too.

@KexinFeng
Copy link
Contributor

KexinFeng commented Sep 18, 2023

The feature is added now in PR #2776 . Welcome to give it a try or make further optimization and contribution.

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

3 participants