@huggingface/tasks/src/tasks/image-segmentation/spec/input.json

List of ML tasks for huggingface.co/tasks

{
	"$id": "/inference/schemas/image-segmentation/input.json",
	"$schema": "http://json-schema.org/draft-06/schema#",
	"description": "Inputs for Image Segmentation inference",
	"title": "ImageSegmentationInput",
	"type": "object",
	"properties": {
		"inputs": {
			"type": "string",
			"description": "The input image data as a base64-encoded string. If no `parameters` are provided, you can also provide the image data as a raw bytes payload.",
			"comment": "type=binary"
		},
		"parameters": {
			"description": "Additional inference parameters for Image Segmentation",
			"$ref": "#/$defs/ImageSegmentationParameters"
		}
	},
	"$defs": {
		"ImageSegmentationParameters": {
			"title": "ImageSegmentationParameters",
			"type": "object",
			"properties": {
				"mask_threshold": {
					"type": "number",
					"description": "Threshold to use when turning the predicted masks into binary values."
				},
				"overlap_mask_area_threshold": {
					"type": "number",
					"description": "Mask overlap threshold to eliminate small, disconnected segments."
				},
				"subtask": {
					"title": "ImageSegmentationSubtask",
					"type": "string",
					"description": "Segmentation task to be performed, depending on model capabilities.",
					"enum": ["instance", "panoptic", "semantic"]
				},
				"threshold": {
					"type": "number",
					"description": "Probability threshold to filter out predicted masks."
				}
			}
		}
	},
	"required": ["inputs"]
}