Add config.json

config.json

@@ -0,0 +1,49 @@
+{
+  "_name_or_path": "best",
+  "architectures": [
+    "YOLOSegmentationPipeline"
+  ],
+  "auto_map": {
+    "AutoModel": "modeling_yolo.YOLOSegmentationPipeline",
+    "AutoModelForImageSegmentation": "modeling_yolo.YOLOSegmentationPipeline"
+  },
+  "framework": "pytorch",
+  "id2label": {
+    "0": "streak"
+  },
+  "label2id": {
+    "streak": 0
+  },
+  "labels": [
+    "streak"
+  ],
+  "library_name": "transformers",
+  "metrics": {
+    "map50": 0.87504,
+    "map50_95": 0.61945
+  },
+  "model_type": "yolo",
+  "pipeline_tag": "image-segmentation",
+  "feature_extractor_type": "YOLOImageProcessor",
+  "processor_class": "YOLOImageProcessor",
+  "image_processor_type": "YOLOImageProcessor",
+  "inference": {
+    "parameters": {
+      "threshold": 0.5,
+      "iou_threshold": 0.5,
+      "mask_threshold": 0.5
+    }
+  },
+  "task_specific_params": {
+    "image-segmentation": {
+      "model_class": "YOLOSegmentationPipeline"
+    }
+  },
+  "tags": [
+    "object-detection",
+    "image-segmentation",
+    "yolo",
+    "endpoints_compatible",
+    "has_space"
+  ]
+}