| { |
| "_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" |
| ] |
| } |
