Skip to content

Commit

Permalink
[onnxruntime] Adds yolov8n pose model for OnnxRuntime (#3309)
Browse files Browse the repository at this point in the history
  • Loading branch information
frankfliu authored Jul 9, 2024
1 parent 6910b0a commit 49f0573
Show file tree
Hide file tree
Showing 2 changed files with 41 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -32,6 +32,7 @@ public class OrtModelZoo extends ModelZoo {
addModel(REPOSITORY.model(CV.IMAGE_CLASSIFICATION, GROUP_ID, "resnet", "0.0.1"));
addModel(REPOSITORY.model(CV.OBJECT_DETECTION, GROUP_ID, "yolo5s", "0.0.1"));
addModel(REPOSITORY.model(CV.OBJECT_DETECTION, GROUP_ID, "yolov8n", "0.0.1"));
addModel(REPOSITORY.model(CV.POSE_ESTIMATION, GROUP_ID, "yolov8n-pose", "0.0.1"));
addModel(REPOSITORY.model(Tabular.SOFTMAX_REGRESSION, GROUP_ID, "iris_flowers", "0.0.1"));
}

Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,40 @@
{
"metadataVersion": "0.2",
"resourceType": "model",
"application": "cv/pose_estimation",
"groupId": "ai.djl.onnxruntime",
"artifactId": "yolov8n-pose",
"name": "Yolov8n pose",
"description": "Yolov8n Pose Estimation",
"website": "http://www.djl.ai/engines/onnxruntime/onnxruntime-engine",
"licenses": {
"license": {
"name": "The Apache License, Version 2.0",
"url": "https://www.apache.org/licenses/LICENSE-2.0"
}
},
"artifacts": [
{
"version": "0.0.1",
"snapshot": false,
"name": "yolov8n-pose",
"properties": {
},
"arguments": {
"width": 640,
"height": 640,
"resize": true,
"threshold": 0.25,
"translatorFactory": "ai.djl.modality.cv.translator.YoloPoseTranslatorFactory"
},
"files": {
"model": {
"uri": "0.0.1/yolov8n-pose.zip",
"sha1Hash": "8b968172ad2dd80a40fd16a276274ee8e2a2ff44",
"name": "",
"size": 11535121
}
}
}
]
}

0 comments on commit 49f0573

Please sign in to comment.