Skip to content

Commit

Permalink
[onnxruntime] Adds yolov8n-seg model to onnxruntime model zoo (#3310)
Browse files Browse the repository at this point in the history
  • Loading branch information
frankfliu authored Jul 11, 2024
1 parent c6e3cdd commit 90680b3
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 @@ -30,6 +30,7 @@ public class OrtModelZoo extends ModelZoo {

OrtModelZoo() {
addModel(REPOSITORY.model(CV.IMAGE_CLASSIFICATION, GROUP_ID, "resnet", "0.0.1"));
addModel(REPOSITORY.model(CV.INSTANCE_SEGMENTATION, GROUP_ID, "yolov8n-seg", "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"));
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,40 @@
{
"metadataVersion": "0.2",
"resourceType": "model",
"application": "cv/instance_segmentation",
"groupId": "ai.djl.onnxruntime",
"artifactId": "yolov8n-seg",
"name": "Yolov8n seg",
"description": "Yolov8n Instance Segmentation",
"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-seg",
"properties": {
},
"arguments": {
"width": 640,
"height": 640,
"resize": true,
"threshold": 0.25,
"translatorFactory": "ai.djl.modality.cv.translator.YoloSegmentationTranslatorFactory"
},
"files": {
"model": {
"uri": "0.0.1/yolov8n-seg.zip",
"sha1Hash": "dd7411362de950dcce003ecca847fd591757a364",
"name": "",
"size": 11913575
}
}
}
]
}

0 comments on commit 90680b3

Please sign in to comment.