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