Upload folder using huggingface_hub
Browse files- .gitattributes +1 -0
- README.md +70 -0
- config.json +24 -0
- rfdetr_seg_medium.bin +3 -0
- rfdetr_seg_medium.xml +0 -0
.gitattributes
CHANGED
|
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
| 36 |
+
*.gif filter=lfs diff=lfs merge=lfs -text
|
README.md
ADDED
|
@@ -0,0 +1,70 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: apache-2.0
|
| 3 |
+
tags:
|
| 4 |
+
- image-segmentation
|
| 5 |
+
- instance-segmentation
|
| 6 |
+
- vision
|
| 7 |
+
---
|
| 8 |
+
|
| 9 |
+
# rfdetr_seg_medium-fp16-ov
|
| 10 |
+
|
| 11 |
+
- Model creator: [Geti™](https://github.com/open-edge-platform/geti)
|
| 12 |
+
- Original model: [RF-DETR Segmentation Medium](https://github.com/roboflow/rf-detr)
|
| 13 |
+
|
| 14 |
+
## Description
|
| 15 |
+
|
| 16 |
+
This is a [Geti™](https://github.com/open-edge-platform/geti) version of [RF-DETR Segmentation Medium](https://github.com/roboflow/rf-detr) model converted to the [OpenVINO™ IR](https://docs.openvino.ai/2026/documentation/openvino-ir-format.html) (Intermediate Representation) format with weights compressed to FP16.
|
| 17 |
+
|
| 18 |
+
To fine-tune your model with a custom dataset, you can use Geti™ to annotate data, perform fine-tuning, and export the resulting model.
|
| 19 |
+
|
| 20 |
+
## Compatibility
|
| 21 |
+
|
| 22 |
+
The provided OpenVINO™ IR model is compatible with:
|
| 23 |
+
|
| 24 |
+
- OpenVINO version 2026.1.0 and higher
|
| 25 |
+
- Model API 0.4.0 and higher
|
| 26 |
+
|
| 27 |
+
## Running Model Inference with [Model API](https://github.com/open-edge-platform/model_api)
|
| 28 |
+
|
| 29 |
+
1. Install required packages:
|
| 30 |
+
|
| 31 |
+
```sh
|
| 32 |
+
pip install openvino-model-api[huggingface]
|
| 33 |
+
```
|
| 34 |
+
|
| 35 |
+
<!-- markdownlint-disable MD029 -->
|
| 36 |
+
|
| 37 |
+
2. Run model inference:
|
| 38 |
+
|
| 39 |
+
```python
|
| 40 |
+
import cv2
|
| 41 |
+
from model_api.models import Model
|
| 42 |
+
from model_api.visualizer import Visualizer
|
| 43 |
+
|
| 44 |
+
# 1. Load model
|
| 45 |
+
model = Model.from_pretrained("OpenVINO/rfdetr_seg_medium-fp16-ov")
|
| 46 |
+
|
| 47 |
+
# 2. Load image
|
| 48 |
+
image = cv2.imread("image.jpg")
|
| 49 |
+
|
| 50 |
+
# 3. Run inference
|
| 51 |
+
result = model(image)
|
| 52 |
+
|
| 53 |
+
# 4. Visualize and save results
|
| 54 |
+
vis = Visualizer().render(image, result)
|
| 55 |
+
cv2.imwrite("output.jpg", vis)
|
| 56 |
+
```
|
| 57 |
+
|
| 58 |
+
For more examples and possible optimizations, refer to the [Model API Documentation](https://open-edge-platform.github.io/model_api/latest/).
|
| 59 |
+
|
| 60 |
+
## Limitations
|
| 61 |
+
|
| 62 |
+
Check the [original model documentation](https://github.com/roboflow/rf-detr) for limitations.
|
| 63 |
+
|
| 64 |
+
## Legal information
|
| 65 |
+
|
| 66 |
+
The original model is distributed under the [Apache-2.0](https://roboflow.com/licensing) license. More details can be found in the [original model repository](https://github.com/roboflow/rf-detr).
|
| 67 |
+
|
| 68 |
+
## Disclaimer
|
| 69 |
+
|
| 70 |
+
Intel is committed to respecting human rights and avoiding causing or contributing to adverse impacts on human rights. See [Intel's Global Human Rights Principles](https://www.intel.com/content/dam/www/central-libraries/us/en/documents/policy-human-rights.pdf). Intel's products and software are intended only to be used in applications that do not cause or contribute to adverse impacts on human rights.
|
config.json
ADDED
|
@@ -0,0 +1,24 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"confidence_threshold": "0.05",
|
| 3 |
+
"getitune_version": "0.1.0",
|
| 4 |
+
"input_dtype": "u8",
|
| 5 |
+
"intensity_min_value": "0.0",
|
| 6 |
+
"intensity_mode": "scale_to_unit",
|
| 7 |
+
"intensity_percentile_high": "99.0",
|
| 8 |
+
"intensity_percentile_low": "1.0",
|
| 9 |
+
"intensity_scale_factor": "1.0",
|
| 10 |
+
"iou_threshold": "0.8",
|
| 11 |
+
"label_ids": "None 0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90",
|
| 12 |
+
"label_info": "{\"label_names\": [\"getitune_empty_lbl\", \"label_0\", \"label_1\", \"label_2\", \"label_3\", \"label_4\", \"label_5\", \"label_6\", \"label_7\", \"label_8\", \"label_9\", \"label_10\", \"label_11\", \"label_12\", \"label_13\", \"label_14\", \"label_15\", \"label_16\", \"label_17\", \"label_18\", \"label_19\", \"label_20\", \"label_21\", \"label_22\", \"label_23\", \"label_24\", \"label_25\", \"label_26\", \"label_27\", \"label_28\", \"label_29\", \"label_30\", \"label_31\", \"label_32\", \"label_33\", \"label_34\", \"label_35\", \"label_36\", \"label_37\", \"label_38\", \"label_39\", \"label_40\", \"label_41\", \"label_42\", \"label_43\", \"label_44\", \"label_45\", \"label_46\", \"label_47\", \"label_48\", \"label_49\", \"label_50\", \"label_51\", \"label_52\", \"label_53\", \"label_54\", \"label_55\", \"label_56\", \"label_57\", \"label_58\", \"label_59\", \"label_60\", \"label_61\", \"label_62\", \"label_63\", \"label_64\", \"label_65\", \"label_66\", \"label_67\", \"label_68\", \"label_69\", \"label_70\", \"label_71\", \"label_72\", \"label_73\", \"label_74\", \"label_75\", \"label_76\", \"label_77\", \"label_78\", \"label_79\", \"label_80\", \"label_81\", \"label_82\", \"label_83\", \"label_84\", \"label_85\", \"label_86\", \"label_87\", \"label_88\", \"label_89\", \"label_90\"], \"label_ids\": [\"None\", \"0\", \"1\", \"2\", \"3\", \"4\", \"5\", \"6\", \"7\", \"8\", \"9\", \"10\", \"11\", \"12\", \"13\", \"14\", \"15\", \"16\", \"17\", \"18\", \"19\", \"20\", \"21\", \"22\", \"23\", \"24\", \"25\", \"26\", \"27\", \"28\", \"29\", \"30\", \"31\", \"32\", \"33\", \"34\", \"35\", \"36\", \"37\", \"38\", \"39\", \"40\", \"41\", \"42\", \"43\", \"44\", \"45\", \"46\", \"47\", \"48\", \"49\", \"50\", \"51\", \"52\", \"53\", \"54\", \"55\", \"56\", \"57\", \"58\", \"59\", \"60\", \"61\", \"62\", \"63\", \"64\", \"65\", \"66\", \"67\", \"68\", \"69\", \"70\", \"71\", \"72\", \"73\", \"74\", \"75\", \"76\", \"77\", \"78\", \"79\", \"80\", \"81\", \"82\", \"83\", \"84\", \"85\", \"86\", \"87\", \"88\", \"89\", \"90\"], \"label_groups\": [[\"getitune_empty_lbl\", \"label_0\", \"label_1\", \"label_2\", \"label_3\", \"label_4\", \"label_5\", \"label_6\", \"label_7\", \"label_8\", \"label_9\", \"label_10\", \"label_11\", \"label_12\", \"label_13\", \"label_14\", \"label_15\", \"label_16\", \"label_17\", \"label_18\", \"label_19\", \"label_20\", \"label_21\", \"label_22\", \"label_23\", \"label_24\", \"label_25\", \"label_26\", \"label_27\", \"label_28\", \"label_29\", \"label_30\", \"label_31\", \"label_32\", \"label_33\", \"label_34\", \"label_35\", \"label_36\", \"label_37\", \"label_38\", \"label_39\", \"label_40\", \"label_41\", \"label_42\", \"label_43\", \"label_44\", \"label_45\", \"label_46\", \"label_47\", \"label_48\", \"label_49\", \"label_50\", \"label_51\", \"label_52\", \"label_53\", \"label_54\", \"label_55\", \"label_56\", \"label_57\", \"label_58\", \"label_59\", \"label_60\", \"label_61\", \"label_62\", \"label_63\", \"label_64\", \"label_65\", \"label_66\", \"label_67\", \"label_68\", \"label_69\", \"label_70\", \"label_71\", \"label_72\", \"label_73\", \"label_74\", \"label_75\", \"label_76\", \"label_77\", \"label_78\", \"label_79\", \"label_80\", \"label_81\", \"label_82\", \"label_83\", \"label_84\", \"label_85\", \"label_86\", \"label_87\", \"label_88\", \"label_89\", \"label_90\"]]}",
|
| 13 |
+
"labels": "__background__ person bicycle car motorcycle airplane bus train truck boat traffic_light fire_hydrant N/A stop_sign parking_meter bench bird cat dog horse sheep cow elephant bear zebra giraffe N/A backpack umbrella N/A N/A handbag tie suitcase frisbee skis snowboard sports_ball kite baseball_bat baseball_glove skateboard surfboard tennis_racket bottle N/A wine_glass cup fork knife spoon bowl banana apple sandwich orange broccoli carrot hot_dog pizza donut cake chair couch potted_plant bed N/A dining_table N/A N/A toilet N/A tv laptop mouse remote keyboard cell_phone microwave oven toaster sink refrigerator N/A book clock vase scissors teddy_bear hair_drier toothbrush",
|
| 14 |
+
"mean_values": "0.0 0.0 0.0",
|
| 15 |
+
"model_name": "rfdetr_seg_m",
|
| 16 |
+
"model_type": "DETRInstSeg",
|
| 17 |
+
"nms_execute": "True",
|
| 18 |
+
"optimization_config": "{\"model_type\": \"transformer\"}",
|
| 19 |
+
"pad_value": "0",
|
| 20 |
+
"resize_type": "standard",
|
| 21 |
+
"reverse_input_channels": "False",
|
| 22 |
+
"scale_values": "1.0 1.0 1.0",
|
| 23 |
+
"task_type": "instance_segmentation"
|
| 24 |
+
}
|
rfdetr_seg_medium.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:db2f86d3a773f383e501106eb366bdbab4a2afd5bb81e4505b982c187e4919ce
|
| 3 |
+
size 64776486
|
rfdetr_seg_medium.xml
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|