toposync-ext-vision 0.1.3__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- toposync_ext_vision/__init__.py +3 -0
- toposync_ext_vision/extension.json +6 -0
- toposync_ext_vision/manifests/rfdetr_det_medium.json +63 -0
- toposync_ext_vision/manifests/rfdetr_det_nano.json +63 -0
- toposync_ext_vision/manifests/rfdetr_det_small.json +63 -0
- toposync_ext_vision/manifests/rtmdet_det_medium.json +64 -0
- toposync_ext_vision/manifests/rtmdet_det_small.json +64 -0
- toposync_ext_vision/manifests/rtmdet_det_tiny.json +64 -0
- toposync_ext_vision/manifests/rtmdet_ins_medium.json +60 -0
- toposync_ext_vision/manifests/rtmdet_ins_small.json +60 -0
- toposync_ext_vision/manifests/rtmdet_ins_tiny.json +60 -0
- toposync_ext_vision/pipelines/__init__.py +99 -0
- toposync_ext_vision/pipelines/operators.py +247 -0
- toposync_ext_vision/pipelines/schemas.py +201 -0
- toposync_ext_vision/plugin.py +88 -0
- toposync_ext_vision/processing/__init__.py +52 -0
- toposync_ext_vision/processing/artifact_helpers.py +413 -0
- toposync_ext_vision/processing/contracts.py +325 -0
- toposync_ext_vision/processing/diagnostics.py +237 -0
- toposync_ext_vision/processing/parsers/__init__.py +21 -0
- toposync_ext_vision/processing/parsers/generic_onnx_boxes_parser.py +141 -0
- toposync_ext_vision/processing/parsers/generic_segmentation_masks_parser.py +156 -0
- toposync_ext_vision/processing/parsers/image_classification_logits_parser.py +102 -0
- toposync_ext_vision/processing/parsers/rfdetr_parser.py +97 -0
- toposync_ext_vision/processing/parsers/rtmdet_ins_parser.py +194 -0
- toposync_ext_vision/processing/parsers/rtmdet_parser.py +112 -0
- toposync_ext_vision/processing/runtime_backends/__init__.py +27 -0
- toposync_ext_vision/processing/runtime_backends/onnxruntime_backend.py +457 -0
- toposync_ext_vision/processing/runtime_upgrades.py +211 -0
- toposync_ext_vision/processing/tasks/__init__.py +13 -0
- toposync_ext_vision/processing/tasks/classification.py +170 -0
- toposync_ext_vision/processing/tasks/detection.py +267 -0
- toposync_ext_vision/processing/tasks/pose.py +308 -0
- toposync_ext_vision/processing/tasks/segmentation.py +506 -0
- toposync_ext_vision/processing/tasks/tracking.py +759 -0
- toposync_ext_vision/processing/trackers/__init__.py +59 -0
- toposync_ext_vision/processing/trackers/norfair_tracker.py +195 -0
- toposync_ext_vision/processing/trackers/simple_iou_kalman.py +349 -0
- toposync_ext_vision/registry/__init__.py +97 -0
- toposync_ext_vision/registry/artifact_upload.py +104 -0
- toposync_ext_vision/registry/builtin_data.py +125 -0
- toposync_ext_vision/registry/custom_onnx.py +756 -0
- toposync_ext_vision/registry/huggingface.py +580 -0
- toposync_ext_vision/registry/huggingface_recipes.py +454 -0
- toposync_ext_vision/registry/installer.py +789 -0
- toposync_ext_vision/registry/local_build.py +922 -0
- toposync_ext_vision/registry/manifests.py +459 -0
- toposync_ext_vision/registry/model_store.py +183 -0
- toposync_ext_vision/registry/recommendations.py +540 -0
- toposync_ext_vision-0.1.3.dist-info/METADATA +114 -0
- toposync_ext_vision-0.1.3.dist-info/RECORD +54 -0
- toposync_ext_vision-0.1.3.dist-info/WHEEL +4 -0
- toposync_ext_vision-0.1.3.dist-info/entry_points.txt +2 -0
- toposync_ext_vision-0.1.3.dist-info/licenses/LICENSE +21 -0
|
@@ -0,0 +1,63 @@
|
|
|
1
|
+
{
|
|
2
|
+
"model_id": "rfdetr_det_medium",
|
|
3
|
+
"display_name": "RF-DETR Medium",
|
|
4
|
+
"task": "detection",
|
|
5
|
+
"runtime": "onnxruntime",
|
|
6
|
+
"artifact_format": "onnx",
|
|
7
|
+
"artifact_path": "../models/rfdetr/rfdetr_det_medium.inference_model.onnx",
|
|
8
|
+
"sha256": "",
|
|
9
|
+
"input": {
|
|
10
|
+
"width": 576,
|
|
11
|
+
"height": 576,
|
|
12
|
+
"color_order": "rgb",
|
|
13
|
+
"layout": "nchw",
|
|
14
|
+
"resize_mode": "stretch",
|
|
15
|
+
"tensor_name": "input",
|
|
16
|
+
"normalization": {
|
|
17
|
+
"mean": [123.675, 116.28, 103.53],
|
|
18
|
+
"std": [58.395, 57.12, 57.375]
|
|
19
|
+
}
|
|
20
|
+
},
|
|
21
|
+
"postprocess": {
|
|
22
|
+
"type": "rfdetr_detr",
|
|
23
|
+
"output_name": "dets",
|
|
24
|
+
"label_output_name": "labels",
|
|
25
|
+
"box_format": "xyxy01",
|
|
26
|
+
"confidence_threshold_default": 0.4,
|
|
27
|
+
"iou_threshold_default": 0.65
|
|
28
|
+
},
|
|
29
|
+
"classes": {
|
|
30
|
+
"source": "coco80"
|
|
31
|
+
},
|
|
32
|
+
"license": {
|
|
33
|
+
"code_license": "Apache-2.0",
|
|
34
|
+
"weights_license": "Apache-2.0 (Apache-designated RF-DETR open weights)",
|
|
35
|
+
"dataset_notes": "COCO-pretrained open weights from the official RF-DETR package; keep Apache notices and upstream attribution.",
|
|
36
|
+
"redistribution_allowed": false,
|
|
37
|
+
"commercial_use_status": "apache_notice_required",
|
|
38
|
+
"official_build_allowed": true
|
|
39
|
+
},
|
|
40
|
+
"acquisition": {
|
|
41
|
+
"mode": "local_build_assisted",
|
|
42
|
+
"artifact_source": "checkpoint_export_required",
|
|
43
|
+
"guide_url": "https://github.com/roboflow/rf-detr",
|
|
44
|
+
"export_guide_url": "https://rfdetr.roboflow.com/learn/export/",
|
|
45
|
+
"source_url": "https://github.com/roboflow/rf-detr",
|
|
46
|
+
"checkpoint_url": "https://storage.googleapis.com/rfdetr/medium_coco/checkpoint_best_regular.pth",
|
|
47
|
+
"paper_url": "https://arxiv.org/abs/2511.09554",
|
|
48
|
+
"builder_backend": "host_python",
|
|
49
|
+
"supported_platforms": ["linux", "darwin", "windows"],
|
|
50
|
+
"explicit_consent_required": true
|
|
51
|
+
},
|
|
52
|
+
"hardware_profiles": {
|
|
53
|
+
"cpu": true,
|
|
54
|
+
"cuda": true,
|
|
55
|
+
"openvino": false,
|
|
56
|
+
"mps": true
|
|
57
|
+
},
|
|
58
|
+
"recommended_profiles": ["cuda_quality", "cpu_balanced"],
|
|
59
|
+
"notes": [
|
|
60
|
+
"Higher-quality RF-DETR option with official ONNX export and Apache-designated open weights.",
|
|
61
|
+
"Best fit when you want the easiest first-party portable path without dropping quality."
|
|
62
|
+
]
|
|
63
|
+
}
|
|
@@ -0,0 +1,63 @@
|
|
|
1
|
+
{
|
|
2
|
+
"model_id": "rfdetr_det_nano",
|
|
3
|
+
"display_name": "RF-DETR Nano",
|
|
4
|
+
"task": "detection",
|
|
5
|
+
"runtime": "onnxruntime",
|
|
6
|
+
"artifact_format": "onnx",
|
|
7
|
+
"artifact_path": "../models/rfdetr/rfdetr_det_nano.inference_model.onnx",
|
|
8
|
+
"sha256": "",
|
|
9
|
+
"input": {
|
|
10
|
+
"width": 384,
|
|
11
|
+
"height": 384,
|
|
12
|
+
"color_order": "rgb",
|
|
13
|
+
"layout": "nchw",
|
|
14
|
+
"resize_mode": "stretch",
|
|
15
|
+
"tensor_name": "input",
|
|
16
|
+
"normalization": {
|
|
17
|
+
"mean": [123.675, 116.28, 103.53],
|
|
18
|
+
"std": [58.395, 57.12, 57.375]
|
|
19
|
+
}
|
|
20
|
+
},
|
|
21
|
+
"postprocess": {
|
|
22
|
+
"type": "rfdetr_detr",
|
|
23
|
+
"output_name": "dets",
|
|
24
|
+
"label_output_name": "labels",
|
|
25
|
+
"box_format": "xyxy01",
|
|
26
|
+
"confidence_threshold_default": 0.4,
|
|
27
|
+
"iou_threshold_default": 0.65
|
|
28
|
+
},
|
|
29
|
+
"classes": {
|
|
30
|
+
"source": "coco80"
|
|
31
|
+
},
|
|
32
|
+
"license": {
|
|
33
|
+
"code_license": "Apache-2.0",
|
|
34
|
+
"weights_license": "Apache-2.0 (Apache-designated RF-DETR open weights)",
|
|
35
|
+
"dataset_notes": "COCO-pretrained open weights from the official RF-DETR package; keep Apache notices and upstream attribution.",
|
|
36
|
+
"redistribution_allowed": false,
|
|
37
|
+
"commercial_use_status": "apache_notice_required",
|
|
38
|
+
"official_build_allowed": true
|
|
39
|
+
},
|
|
40
|
+
"acquisition": {
|
|
41
|
+
"mode": "local_build_assisted",
|
|
42
|
+
"artifact_source": "checkpoint_export_required",
|
|
43
|
+
"guide_url": "https://github.com/roboflow/rf-detr",
|
|
44
|
+
"export_guide_url": "https://rfdetr.roboflow.com/learn/export/",
|
|
45
|
+
"source_url": "https://github.com/roboflow/rf-detr",
|
|
46
|
+
"checkpoint_url": "https://storage.googleapis.com/rfdetr/nano_coco/checkpoint_best_regular.pth",
|
|
47
|
+
"paper_url": "https://arxiv.org/abs/2511.09554",
|
|
48
|
+
"builder_backend": "host_python",
|
|
49
|
+
"supported_platforms": ["linux", "darwin", "windows"],
|
|
50
|
+
"explicit_consent_required": true
|
|
51
|
+
},
|
|
52
|
+
"hardware_profiles": {
|
|
53
|
+
"cpu": true,
|
|
54
|
+
"cuda": true,
|
|
55
|
+
"openvino": false,
|
|
56
|
+
"mps": true
|
|
57
|
+
},
|
|
58
|
+
"recommended_profiles": ["cpu_low", "cuda_low"],
|
|
59
|
+
"notes": [
|
|
60
|
+
"Portable RF-DETR entry point with official ONNX export in the upstream package.",
|
|
61
|
+
"Strong quality for a lightweight cross-platform assisted local build path."
|
|
62
|
+
]
|
|
63
|
+
}
|
|
@@ -0,0 +1,63 @@
|
|
|
1
|
+
{
|
|
2
|
+
"model_id": "rfdetr_det_small",
|
|
3
|
+
"display_name": "RF-DETR Small",
|
|
4
|
+
"task": "detection",
|
|
5
|
+
"runtime": "onnxruntime",
|
|
6
|
+
"artifact_format": "onnx",
|
|
7
|
+
"artifact_path": "../models/rfdetr/rfdetr_det_small.inference_model.onnx",
|
|
8
|
+
"sha256": "",
|
|
9
|
+
"input": {
|
|
10
|
+
"width": 512,
|
|
11
|
+
"height": 512,
|
|
12
|
+
"color_order": "rgb",
|
|
13
|
+
"layout": "nchw",
|
|
14
|
+
"resize_mode": "stretch",
|
|
15
|
+
"tensor_name": "input",
|
|
16
|
+
"normalization": {
|
|
17
|
+
"mean": [123.675, 116.28, 103.53],
|
|
18
|
+
"std": [58.395, 57.12, 57.375]
|
|
19
|
+
}
|
|
20
|
+
},
|
|
21
|
+
"postprocess": {
|
|
22
|
+
"type": "rfdetr_detr",
|
|
23
|
+
"output_name": "dets",
|
|
24
|
+
"label_output_name": "labels",
|
|
25
|
+
"box_format": "xyxy01",
|
|
26
|
+
"confidence_threshold_default": 0.4,
|
|
27
|
+
"iou_threshold_default": 0.65
|
|
28
|
+
},
|
|
29
|
+
"classes": {
|
|
30
|
+
"source": "coco80"
|
|
31
|
+
},
|
|
32
|
+
"license": {
|
|
33
|
+
"code_license": "Apache-2.0",
|
|
34
|
+
"weights_license": "Apache-2.0 (Apache-designated RF-DETR open weights)",
|
|
35
|
+
"dataset_notes": "COCO-pretrained open weights from the official RF-DETR package; keep Apache notices and upstream attribution.",
|
|
36
|
+
"redistribution_allowed": false,
|
|
37
|
+
"commercial_use_status": "apache_notice_required",
|
|
38
|
+
"official_build_allowed": true
|
|
39
|
+
},
|
|
40
|
+
"acquisition": {
|
|
41
|
+
"mode": "local_build_assisted",
|
|
42
|
+
"artifact_source": "checkpoint_export_required",
|
|
43
|
+
"guide_url": "https://github.com/roboflow/rf-detr",
|
|
44
|
+
"export_guide_url": "https://rfdetr.roboflow.com/learn/export/",
|
|
45
|
+
"source_url": "https://github.com/roboflow/rf-detr",
|
|
46
|
+
"checkpoint_url": "https://storage.googleapis.com/rfdetr/small_coco/checkpoint_best_regular.pth",
|
|
47
|
+
"paper_url": "https://arxiv.org/abs/2511.09554",
|
|
48
|
+
"builder_backend": "host_python",
|
|
49
|
+
"supported_platforms": ["linux", "darwin", "windows"],
|
|
50
|
+
"explicit_consent_required": true
|
|
51
|
+
},
|
|
52
|
+
"hardware_profiles": {
|
|
53
|
+
"cpu": true,
|
|
54
|
+
"cuda": true,
|
|
55
|
+
"openvino": false,
|
|
56
|
+
"mps": true
|
|
57
|
+
},
|
|
58
|
+
"recommended_profiles": ["cpu_balanced", "cuda_balanced"],
|
|
59
|
+
"notes": [
|
|
60
|
+
"Balanced RF-DETR option with clearer cross-platform packaging than RTMDet/MMDeploy.",
|
|
61
|
+
"Good one-click candidate when you want strong COCO quality without a container-only builder."
|
|
62
|
+
]
|
|
63
|
+
}
|
|
@@ -0,0 +1,64 @@
|
|
|
1
|
+
{
|
|
2
|
+
"model_id": "rtmdet_det_medium",
|
|
3
|
+
"display_name": "RTMDet Medium",
|
|
4
|
+
"task": "detection",
|
|
5
|
+
"runtime": "onnxruntime",
|
|
6
|
+
"artifact_format": "onnx",
|
|
7
|
+
"artifact_path": "../models/rtmdet/rtmdet_det_medium.end2end.onnx",
|
|
8
|
+
"sha256": "f89e0b0e87f4cb93260ef62a187dfa71fa4dc5fdaf4173e5eeeaa624498fe18e",
|
|
9
|
+
"input": {
|
|
10
|
+
"width": 640,
|
|
11
|
+
"height": 640,
|
|
12
|
+
"color_order": "bgr",
|
|
13
|
+
"layout": "nchw",
|
|
14
|
+
"resize_mode": "letterbox",
|
|
15
|
+
"pad_value": 114.0,
|
|
16
|
+
"normalization": {
|
|
17
|
+
"mean": [103.53, 116.28, 123.675],
|
|
18
|
+
"std": [57.375, 57.12, 58.395]
|
|
19
|
+
}
|
|
20
|
+
},
|
|
21
|
+
"postprocess": {
|
|
22
|
+
"type": "mmdet_rtmdet",
|
|
23
|
+
"output_name": "dets",
|
|
24
|
+
"label_output_name": "labels",
|
|
25
|
+
"box_format": "xyxy_pixels",
|
|
26
|
+
"confidence_threshold_default": 0.4,
|
|
27
|
+
"iou_threshold_default": 0.65
|
|
28
|
+
},
|
|
29
|
+
"classes": {
|
|
30
|
+
"source": "coco80"
|
|
31
|
+
},
|
|
32
|
+
"license": {
|
|
33
|
+
"code_license": "Apache-2.0",
|
|
34
|
+
"weights_license": "Review required for exported ONNX artifact redistribution.",
|
|
35
|
+
"dataset_notes": "COCO-based weights require explicit redistribution review before bundling.",
|
|
36
|
+
"redistribution_allowed": false,
|
|
37
|
+
"commercial_use_status": "review_required",
|
|
38
|
+
"official_build_allowed": false
|
|
39
|
+
},
|
|
40
|
+
"acquisition": {
|
|
41
|
+
"mode": "guided_upload",
|
|
42
|
+
"artifact_source": "checkpoint_export_required",
|
|
43
|
+
"guide_url": "https://github.com/open-mmlab/mmdetection/blob/main/configs/rtmdet/README.md",
|
|
44
|
+
"export_guide_url": "https://mmdeploy.readthedocs.io/en/v1.2.0/04-supported-codebases/mmdet.html",
|
|
45
|
+
"checkpoint_url": "https://download.openmmlab.com/mmdetection/v3.0/rtmdet/rtmdet_m_8xb32-300e_coco/rtmdet_m_8xb32-300e_coco_20220719_112220-229f527c.pth",
|
|
46
|
+
"config_url": "https://github.com/open-mmlab/mmdetection/blob/main/configs/rtmdet/rtmdet_m_8xb32-300e_coco.py",
|
|
47
|
+
"metafile_url": "https://github.com/open-mmlab/mmdetection/blob/main/configs/rtmdet/metafile.yml",
|
|
48
|
+
"paper_url": "https://arxiv.org/abs/2212.07784",
|
|
49
|
+
"builder_backend": "container_local",
|
|
50
|
+
"supported_platforms": ["linux"],
|
|
51
|
+
"explicit_consent_required": true
|
|
52
|
+
},
|
|
53
|
+
"hardware_profiles": {
|
|
54
|
+
"cpu": true,
|
|
55
|
+
"cuda": true,
|
|
56
|
+
"openvino": true,
|
|
57
|
+
"mps": false
|
|
58
|
+
},
|
|
59
|
+
"recommended_profiles": ["cuda_quality"],
|
|
60
|
+
"notes": [
|
|
61
|
+
"Higher-quality RTMDet option for stronger CPU/GPU machines.",
|
|
62
|
+
"Use when object recall matters more than per-frame latency."
|
|
63
|
+
]
|
|
64
|
+
}
|
|
@@ -0,0 +1,64 @@
|
|
|
1
|
+
{
|
|
2
|
+
"model_id": "rtmdet_det_small",
|
|
3
|
+
"display_name": "RTMDet Small",
|
|
4
|
+
"task": "detection",
|
|
5
|
+
"runtime": "onnxruntime",
|
|
6
|
+
"artifact_format": "onnx",
|
|
7
|
+
"artifact_path": "../models/rtmdet/rtmdet_det_small.end2end.onnx",
|
|
8
|
+
"sha256": "e23a7b1ec6b7d7a2bd82b825e6702b7686af95c711ac1aa59655a3dfddf36c7c",
|
|
9
|
+
"input": {
|
|
10
|
+
"width": 640,
|
|
11
|
+
"height": 640,
|
|
12
|
+
"color_order": "bgr",
|
|
13
|
+
"layout": "nchw",
|
|
14
|
+
"resize_mode": "letterbox",
|
|
15
|
+
"pad_value": 114.0,
|
|
16
|
+
"normalization": {
|
|
17
|
+
"mean": [103.53, 116.28, 123.675],
|
|
18
|
+
"std": [57.375, 57.12, 58.395]
|
|
19
|
+
}
|
|
20
|
+
},
|
|
21
|
+
"postprocess": {
|
|
22
|
+
"type": "mmdet_rtmdet",
|
|
23
|
+
"output_name": "dets",
|
|
24
|
+
"label_output_name": "labels",
|
|
25
|
+
"box_format": "xyxy_pixels",
|
|
26
|
+
"confidence_threshold_default": 0.4,
|
|
27
|
+
"iou_threshold_default": 0.65
|
|
28
|
+
},
|
|
29
|
+
"classes": {
|
|
30
|
+
"source": "coco80"
|
|
31
|
+
},
|
|
32
|
+
"license": {
|
|
33
|
+
"code_license": "Apache-2.0",
|
|
34
|
+
"weights_license": "Review required for exported ONNX artifact redistribution.",
|
|
35
|
+
"dataset_notes": "COCO-based weights require explicit redistribution review before bundling.",
|
|
36
|
+
"redistribution_allowed": false,
|
|
37
|
+
"commercial_use_status": "review_required",
|
|
38
|
+
"official_build_allowed": false
|
|
39
|
+
},
|
|
40
|
+
"acquisition": {
|
|
41
|
+
"mode": "guided_upload",
|
|
42
|
+
"artifact_source": "checkpoint_export_required",
|
|
43
|
+
"guide_url": "https://github.com/open-mmlab/mmdetection/blob/main/configs/rtmdet/README.md",
|
|
44
|
+
"export_guide_url": "https://mmdeploy.readthedocs.io/en/v1.2.0/04-supported-codebases/mmdet.html",
|
|
45
|
+
"checkpoint_url": "https://download.openmmlab.com/mmdetection/v3.0/rtmdet/rtmdet_s_8xb32-300e_coco/rtmdet_s_8xb32-300e_coco_20220905_161602-387a891e.pth",
|
|
46
|
+
"config_url": "https://github.com/open-mmlab/mmdetection/blob/main/configs/rtmdet/rtmdet_s_8xb32-300e_coco.py",
|
|
47
|
+
"metafile_url": "https://github.com/open-mmlab/mmdetection/blob/main/configs/rtmdet/metafile.yml",
|
|
48
|
+
"paper_url": "https://arxiv.org/abs/2212.07784",
|
|
49
|
+
"builder_backend": "container_local",
|
|
50
|
+
"supported_platforms": ["linux"],
|
|
51
|
+
"explicit_consent_required": true
|
|
52
|
+
},
|
|
53
|
+
"hardware_profiles": {
|
|
54
|
+
"cpu": true,
|
|
55
|
+
"cuda": true,
|
|
56
|
+
"openvino": true,
|
|
57
|
+
"mps": false
|
|
58
|
+
},
|
|
59
|
+
"recommended_profiles": ["cpu_balanced", "cuda_balanced", "openvino_balanced"],
|
|
60
|
+
"notes": [
|
|
61
|
+
"Balanced RTMDet option and default recommendation for general COCO detection.",
|
|
62
|
+
"Good trade-off between accuracy and CPU/GPU cost."
|
|
63
|
+
]
|
|
64
|
+
}
|
|
@@ -0,0 +1,64 @@
|
|
|
1
|
+
{
|
|
2
|
+
"model_id": "rtmdet_det_tiny",
|
|
3
|
+
"display_name": "RTMDet Tiny",
|
|
4
|
+
"task": "detection",
|
|
5
|
+
"runtime": "onnxruntime",
|
|
6
|
+
"artifact_format": "onnx",
|
|
7
|
+
"artifact_path": "../models/rtmdet/rtmdet_det_tiny.end2end.onnx",
|
|
8
|
+
"sha256": "7581382585401922c870b3e25080dd1215c7048c632c72499dd5d8e7f9e8fddf",
|
|
9
|
+
"input": {
|
|
10
|
+
"width": 640,
|
|
11
|
+
"height": 640,
|
|
12
|
+
"color_order": "bgr",
|
|
13
|
+
"layout": "nchw",
|
|
14
|
+
"resize_mode": "letterbox",
|
|
15
|
+
"pad_value": 114.0,
|
|
16
|
+
"normalization": {
|
|
17
|
+
"mean": [103.53, 116.28, 123.675],
|
|
18
|
+
"std": [57.375, 57.12, 58.395]
|
|
19
|
+
}
|
|
20
|
+
},
|
|
21
|
+
"postprocess": {
|
|
22
|
+
"type": "mmdet_rtmdet",
|
|
23
|
+
"output_name": "dets",
|
|
24
|
+
"label_output_name": "labels",
|
|
25
|
+
"box_format": "xyxy_pixels",
|
|
26
|
+
"confidence_threshold_default": 0.4,
|
|
27
|
+
"iou_threshold_default": 0.65
|
|
28
|
+
},
|
|
29
|
+
"classes": {
|
|
30
|
+
"source": "coco80"
|
|
31
|
+
},
|
|
32
|
+
"license": {
|
|
33
|
+
"code_license": "Apache-2.0",
|
|
34
|
+
"weights_license": "Review required for exported ONNX artifact redistribution.",
|
|
35
|
+
"dataset_notes": "COCO-based weights require explicit redistribution review before bundling.",
|
|
36
|
+
"redistribution_allowed": false,
|
|
37
|
+
"commercial_use_status": "review_required",
|
|
38
|
+
"official_build_allowed": false
|
|
39
|
+
},
|
|
40
|
+
"acquisition": {
|
|
41
|
+
"mode": "guided_upload",
|
|
42
|
+
"artifact_source": "checkpoint_export_required",
|
|
43
|
+
"guide_url": "https://github.com/open-mmlab/mmdetection/blob/main/configs/rtmdet/README.md",
|
|
44
|
+
"export_guide_url": "https://mmdeploy.readthedocs.io/en/v1.2.0/04-supported-codebases/mmdet.html",
|
|
45
|
+
"checkpoint_url": "https://download.openmmlab.com/mmdetection/v3.0/rtmdet/rtmdet_tiny_8xb32-300e_coco/rtmdet_tiny_8xb32-300e_coco_20220902_112414-78e30dcc.pth",
|
|
46
|
+
"config_url": "https://github.com/open-mmlab/mmdetection/blob/main/configs/rtmdet/rtmdet_tiny_8xb32-300e_coco.py",
|
|
47
|
+
"metafile_url": "https://github.com/open-mmlab/mmdetection/blob/main/configs/rtmdet/metafile.yml",
|
|
48
|
+
"paper_url": "https://arxiv.org/abs/2212.07784",
|
|
49
|
+
"builder_backend": "container_local",
|
|
50
|
+
"supported_platforms": ["linux"],
|
|
51
|
+
"explicit_consent_required": true
|
|
52
|
+
},
|
|
53
|
+
"hardware_profiles": {
|
|
54
|
+
"cpu": true,
|
|
55
|
+
"cuda": true,
|
|
56
|
+
"openvino": true,
|
|
57
|
+
"mps": false
|
|
58
|
+
},
|
|
59
|
+
"recommended_profiles": ["cpu_low", "cuda_low", "openvino_balanced"],
|
|
60
|
+
"notes": [
|
|
61
|
+
"Fastest RTMDet option for CPU-first deployments.",
|
|
62
|
+
"Recommended first smoke-test model for general COCO detection."
|
|
63
|
+
]
|
|
64
|
+
}
|
|
@@ -0,0 +1,60 @@
|
|
|
1
|
+
{
|
|
2
|
+
"model_id": "rtmdet_ins_medium",
|
|
3
|
+
"display_name": "RTMDet-Ins Medium",
|
|
4
|
+
"task": "segmentation",
|
|
5
|
+
"runtime": "onnxruntime",
|
|
6
|
+
"artifact_format": "onnx",
|
|
7
|
+
"artifact_path": "../models/rtmdet/rtmdet_ins_medium.end2end.onnx",
|
|
8
|
+
"sha256": "",
|
|
9
|
+
"input": {
|
|
10
|
+
"width": 640,
|
|
11
|
+
"height": 640,
|
|
12
|
+
"color_order": "bgr",
|
|
13
|
+
"layout": "nchw",
|
|
14
|
+
"resize_mode": "letterbox",
|
|
15
|
+
"pad_value": 114.0,
|
|
16
|
+
"normalization": {
|
|
17
|
+
"mean": [103.53, 116.28, 123.675],
|
|
18
|
+
"std": [57.375, 57.12, 58.395]
|
|
19
|
+
}
|
|
20
|
+
},
|
|
21
|
+
"postprocess": {
|
|
22
|
+
"type": "mmdet_rtmdet_ins",
|
|
23
|
+
"output_name": "dets",
|
|
24
|
+
"label_output_name": "labels",
|
|
25
|
+
"mask_output_name": "masks",
|
|
26
|
+
"box_format": "xyxy_pixels",
|
|
27
|
+
"mask_format": "full_frame_binary",
|
|
28
|
+
"confidence_threshold_default": 0.4,
|
|
29
|
+
"iou_threshold_default": 0.65,
|
|
30
|
+
"polygon_threshold": 0.5
|
|
31
|
+
},
|
|
32
|
+
"classes": {
|
|
33
|
+
"source": "coco80"
|
|
34
|
+
},
|
|
35
|
+
"license": {
|
|
36
|
+
"code_license": "Apache-2.0",
|
|
37
|
+
"weights_license": "Review required for exported ONNX artifact redistribution.",
|
|
38
|
+
"dataset_notes": "COCO-based weights require explicit redistribution review before bundling.",
|
|
39
|
+
"redistribution_allowed": false,
|
|
40
|
+
"commercial_use_status": "review_required",
|
|
41
|
+
"official_build_allowed": false
|
|
42
|
+
},
|
|
43
|
+
"acquisition": {
|
|
44
|
+
"mode": "guided_upload",
|
|
45
|
+
"artifact_source": "checkpoint_export_required",
|
|
46
|
+
"guide_url": "https://github.com/open-mmlab/mmdetection/blob/main/configs/rtmdet/README.md",
|
|
47
|
+
"export_guide_url": "https://mmdeploy.readthedocs.io/en/latest/01-how-to-build/build_from_docker.html"
|
|
48
|
+
},
|
|
49
|
+
"hardware_profiles": {
|
|
50
|
+
"cpu": true,
|
|
51
|
+
"cuda": true,
|
|
52
|
+
"openvino": true,
|
|
53
|
+
"mps": false
|
|
54
|
+
},
|
|
55
|
+
"recommended_profiles": ["cuda_quality"],
|
|
56
|
+
"notes": [
|
|
57
|
+
"Higher-quality RTMDet-Ins option for stronger CPU/GPU machines.",
|
|
58
|
+
"Use when mask quality matters more than throughput."
|
|
59
|
+
]
|
|
60
|
+
}
|
|
@@ -0,0 +1,60 @@
|
|
|
1
|
+
{
|
|
2
|
+
"model_id": "rtmdet_ins_small",
|
|
3
|
+
"display_name": "RTMDet-Ins Small",
|
|
4
|
+
"task": "segmentation",
|
|
5
|
+
"runtime": "onnxruntime",
|
|
6
|
+
"artifact_format": "onnx",
|
|
7
|
+
"artifact_path": "../models/rtmdet/rtmdet_ins_small.end2end.onnx",
|
|
8
|
+
"sha256": "",
|
|
9
|
+
"input": {
|
|
10
|
+
"width": 640,
|
|
11
|
+
"height": 640,
|
|
12
|
+
"color_order": "bgr",
|
|
13
|
+
"layout": "nchw",
|
|
14
|
+
"resize_mode": "letterbox",
|
|
15
|
+
"pad_value": 114.0,
|
|
16
|
+
"normalization": {
|
|
17
|
+
"mean": [103.53, 116.28, 123.675],
|
|
18
|
+
"std": [57.375, 57.12, 58.395]
|
|
19
|
+
}
|
|
20
|
+
},
|
|
21
|
+
"postprocess": {
|
|
22
|
+
"type": "mmdet_rtmdet_ins",
|
|
23
|
+
"output_name": "dets",
|
|
24
|
+
"label_output_name": "labels",
|
|
25
|
+
"mask_output_name": "masks",
|
|
26
|
+
"box_format": "xyxy_pixels",
|
|
27
|
+
"mask_format": "full_frame_binary",
|
|
28
|
+
"confidence_threshold_default": 0.4,
|
|
29
|
+
"iou_threshold_default": 0.65,
|
|
30
|
+
"polygon_threshold": 0.5
|
|
31
|
+
},
|
|
32
|
+
"classes": {
|
|
33
|
+
"source": "coco80"
|
|
34
|
+
},
|
|
35
|
+
"license": {
|
|
36
|
+
"code_license": "Apache-2.0",
|
|
37
|
+
"weights_license": "Review required for exported ONNX artifact redistribution.",
|
|
38
|
+
"dataset_notes": "COCO-based weights require explicit redistribution review before bundling.",
|
|
39
|
+
"redistribution_allowed": false,
|
|
40
|
+
"commercial_use_status": "review_required",
|
|
41
|
+
"official_build_allowed": false
|
|
42
|
+
},
|
|
43
|
+
"acquisition": {
|
|
44
|
+
"mode": "guided_upload",
|
|
45
|
+
"artifact_source": "checkpoint_export_required",
|
|
46
|
+
"guide_url": "https://github.com/open-mmlab/mmdetection/blob/main/configs/rtmdet/README.md",
|
|
47
|
+
"export_guide_url": "https://mmdeploy.readthedocs.io/en/latest/01-how-to-build/build_from_docker.html"
|
|
48
|
+
},
|
|
49
|
+
"hardware_profiles": {
|
|
50
|
+
"cpu": true,
|
|
51
|
+
"cuda": true,
|
|
52
|
+
"openvino": true,
|
|
53
|
+
"mps": false
|
|
54
|
+
},
|
|
55
|
+
"recommended_profiles": ["cpu_balanced", "cuda_balanced", "openvino_balanced"],
|
|
56
|
+
"notes": [
|
|
57
|
+
"Balanced default recommendation for real instance segmentation on COCO.",
|
|
58
|
+
"Good trade-off between mask quality and per-frame latency."
|
|
59
|
+
]
|
|
60
|
+
}
|
|
@@ -0,0 +1,60 @@
|
|
|
1
|
+
{
|
|
2
|
+
"model_id": "rtmdet_ins_tiny",
|
|
3
|
+
"display_name": "RTMDet-Ins Tiny",
|
|
4
|
+
"task": "segmentation",
|
|
5
|
+
"runtime": "onnxruntime",
|
|
6
|
+
"artifact_format": "onnx",
|
|
7
|
+
"artifact_path": "../models/rtmdet/rtmdet_ins_tiny.end2end.onnx",
|
|
8
|
+
"sha256": "",
|
|
9
|
+
"input": {
|
|
10
|
+
"width": 640,
|
|
11
|
+
"height": 640,
|
|
12
|
+
"color_order": "bgr",
|
|
13
|
+
"layout": "nchw",
|
|
14
|
+
"resize_mode": "letterbox",
|
|
15
|
+
"pad_value": 114.0,
|
|
16
|
+
"normalization": {
|
|
17
|
+
"mean": [103.53, 116.28, 123.675],
|
|
18
|
+
"std": [57.375, 57.12, 58.395]
|
|
19
|
+
}
|
|
20
|
+
},
|
|
21
|
+
"postprocess": {
|
|
22
|
+
"type": "mmdet_rtmdet_ins",
|
|
23
|
+
"output_name": "dets",
|
|
24
|
+
"label_output_name": "labels",
|
|
25
|
+
"mask_output_name": "masks",
|
|
26
|
+
"box_format": "xyxy_pixels",
|
|
27
|
+
"mask_format": "full_frame_binary",
|
|
28
|
+
"confidence_threshold_default": 0.4,
|
|
29
|
+
"iou_threshold_default": 0.65,
|
|
30
|
+
"polygon_threshold": 0.5
|
|
31
|
+
},
|
|
32
|
+
"classes": {
|
|
33
|
+
"source": "coco80"
|
|
34
|
+
},
|
|
35
|
+
"license": {
|
|
36
|
+
"code_license": "Apache-2.0",
|
|
37
|
+
"weights_license": "Review required for exported ONNX artifact redistribution.",
|
|
38
|
+
"dataset_notes": "COCO-based weights require explicit redistribution review before bundling.",
|
|
39
|
+
"redistribution_allowed": false,
|
|
40
|
+
"commercial_use_status": "review_required",
|
|
41
|
+
"official_build_allowed": false
|
|
42
|
+
},
|
|
43
|
+
"acquisition": {
|
|
44
|
+
"mode": "guided_upload",
|
|
45
|
+
"artifact_source": "checkpoint_export_required",
|
|
46
|
+
"guide_url": "https://github.com/open-mmlab/mmdetection/blob/main/configs/rtmdet/README.md",
|
|
47
|
+
"export_guide_url": "https://mmdeploy.readthedocs.io/en/latest/01-how-to-build/build_from_docker.html"
|
|
48
|
+
},
|
|
49
|
+
"hardware_profiles": {
|
|
50
|
+
"cpu": true,
|
|
51
|
+
"cuda": true,
|
|
52
|
+
"openvino": true,
|
|
53
|
+
"mps": false
|
|
54
|
+
},
|
|
55
|
+
"recommended_profiles": ["cpu_low", "cuda_low", "openvino_balanced"],
|
|
56
|
+
"notes": [
|
|
57
|
+
"Fastest RTMDet-Ins option for CPU-first instance segmentation.",
|
|
58
|
+
"Good first smoke test for real masks on COCO classes."
|
|
59
|
+
]
|
|
60
|
+
}
|