-
Notifications
You must be signed in to change notification settings - Fork 665
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
[pytorch] Adds Yolo11 model to model zoo #3516
Conversation
examples/src/main/java/ai/djl/examples/inference/cv/PoseEstimation.java
Outdated
Show resolved
Hide resolved
|
||
@Test | ||
public void testYolov8Detection() throws ModelException, TranslateException, IOException { | ||
TestRequirements.notGpu(); | ||
|
||
DetectedObjects result = Yolov8Detection.predict(); | ||
DetectedObjects result = YoloDetection.predict(); | ||
|
||
Assert.assertTrue(result.getNumberOfObjects() >= 1); | ||
Classifications.Classification obj = result.best(); |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
The test needs to be updated:
Assert.assertTrue(result.getClassNames().contains("dog"));
Classifications.Classification obj = result.best();
Assert.assertTrue(obj.getProbability() > 0.6);
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Done
Codecov ReportAll modified and coverable lines are covered by tests ✅
❗ Your organization needs to install the Codecov GitHub app to enable full functionality. Additional details and impacted files@@ Coverage Diff @@
## master #3516 +/- ##
============================================
- Coverage 72.52% 72.37% -0.16%
- Complexity 7394 7411 +17
============================================
Files 672 674 +2
Lines 32909 33048 +139
Branches 3501 3523 +22
============================================
+ Hits 23868 23918 +50
- Misses 7400 7497 +97
+ Partials 1641 1633 -8
Flags with carried forward coverage won't be shown. Click here to find out more. ☔ View full report in Codecov by Sentry. |
Description
Brief description of what this PR is about