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.
Files changed (54) hide show
  1. toposync_ext_vision/__init__.py +3 -0
  2. toposync_ext_vision/extension.json +6 -0
  3. toposync_ext_vision/manifests/rfdetr_det_medium.json +63 -0
  4. toposync_ext_vision/manifests/rfdetr_det_nano.json +63 -0
  5. toposync_ext_vision/manifests/rfdetr_det_small.json +63 -0
  6. toposync_ext_vision/manifests/rtmdet_det_medium.json +64 -0
  7. toposync_ext_vision/manifests/rtmdet_det_small.json +64 -0
  8. toposync_ext_vision/manifests/rtmdet_det_tiny.json +64 -0
  9. toposync_ext_vision/manifests/rtmdet_ins_medium.json +60 -0
  10. toposync_ext_vision/manifests/rtmdet_ins_small.json +60 -0
  11. toposync_ext_vision/manifests/rtmdet_ins_tiny.json +60 -0
  12. toposync_ext_vision/pipelines/__init__.py +99 -0
  13. toposync_ext_vision/pipelines/operators.py +247 -0
  14. toposync_ext_vision/pipelines/schemas.py +201 -0
  15. toposync_ext_vision/plugin.py +88 -0
  16. toposync_ext_vision/processing/__init__.py +52 -0
  17. toposync_ext_vision/processing/artifact_helpers.py +413 -0
  18. toposync_ext_vision/processing/contracts.py +325 -0
  19. toposync_ext_vision/processing/diagnostics.py +237 -0
  20. toposync_ext_vision/processing/parsers/__init__.py +21 -0
  21. toposync_ext_vision/processing/parsers/generic_onnx_boxes_parser.py +141 -0
  22. toposync_ext_vision/processing/parsers/generic_segmentation_masks_parser.py +156 -0
  23. toposync_ext_vision/processing/parsers/image_classification_logits_parser.py +102 -0
  24. toposync_ext_vision/processing/parsers/rfdetr_parser.py +97 -0
  25. toposync_ext_vision/processing/parsers/rtmdet_ins_parser.py +194 -0
  26. toposync_ext_vision/processing/parsers/rtmdet_parser.py +112 -0
  27. toposync_ext_vision/processing/runtime_backends/__init__.py +27 -0
  28. toposync_ext_vision/processing/runtime_backends/onnxruntime_backend.py +457 -0
  29. toposync_ext_vision/processing/runtime_upgrades.py +211 -0
  30. toposync_ext_vision/processing/tasks/__init__.py +13 -0
  31. toposync_ext_vision/processing/tasks/classification.py +170 -0
  32. toposync_ext_vision/processing/tasks/detection.py +267 -0
  33. toposync_ext_vision/processing/tasks/pose.py +308 -0
  34. toposync_ext_vision/processing/tasks/segmentation.py +506 -0
  35. toposync_ext_vision/processing/tasks/tracking.py +759 -0
  36. toposync_ext_vision/processing/trackers/__init__.py +59 -0
  37. toposync_ext_vision/processing/trackers/norfair_tracker.py +195 -0
  38. toposync_ext_vision/processing/trackers/simple_iou_kalman.py +349 -0
  39. toposync_ext_vision/registry/__init__.py +97 -0
  40. toposync_ext_vision/registry/artifact_upload.py +104 -0
  41. toposync_ext_vision/registry/builtin_data.py +125 -0
  42. toposync_ext_vision/registry/custom_onnx.py +756 -0
  43. toposync_ext_vision/registry/huggingface.py +580 -0
  44. toposync_ext_vision/registry/huggingface_recipes.py +454 -0
  45. toposync_ext_vision/registry/installer.py +789 -0
  46. toposync_ext_vision/registry/local_build.py +922 -0
  47. toposync_ext_vision/registry/manifests.py +459 -0
  48. toposync_ext_vision/registry/model_store.py +183 -0
  49. toposync_ext_vision/registry/recommendations.py +540 -0
  50. toposync_ext_vision-0.1.3.dist-info/METADATA +114 -0
  51. toposync_ext_vision-0.1.3.dist-info/RECORD +54 -0
  52. toposync_ext_vision-0.1.3.dist-info/WHEEL +4 -0
  53. toposync_ext_vision-0.1.3.dist-info/entry_points.txt +2 -0
  54. toposync_ext_vision-0.1.3.dist-info/licenses/LICENSE +21 -0
@@ -0,0 +1,3 @@
1
+ from __future__ import annotations
2
+
3
+ __all__ = []
@@ -0,0 +1,6 @@
1
+ {
2
+ "schema_version": 1,
3
+ "id": "com.toposync.vision",
4
+ "name": "Vision",
5
+ "version": "0.1.3"
6
+ }
@@ -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
+ }