Florence2-image-Annotator / mellon_pipeline_config.json
YiYiXu's picture
YiYiXu HF Staff
Update mellon_pipeline_config.json
ffb3067 verified
{
"label": "Florence2 Image Annotator Block",
"default_repo": "",
"default_dtype": "bfloat16",
"node_params": {
"custom": {
"params": {
"image": {
"label": "Image",
"type": "image",
"display": "input"
},
"annotation_task": {
"label": "Annotation Task",
"type": "string",
"options": [
"<CAPTION_TO_PHRASE_GROUNDING>",
"<REFERRING_EXPRESSION_SEGMENTATION>",
"<OPEN_VOCABULARY_DETECTION>"
],
"value": "<CAPTION_TO_PHRASE_GROUNDING>"
},
"annotation_prompt": {
"label": "Annotation Prompt",
"type": "string",
"display": "textarea",
"default": ""
},
"annotation_output_type": {
"label": "Annotation Output Type",
"type": "string",
"options": [
"bounding_box",
"mask_overlay",
"mask_image"
],
"value": "mask_image"
},
"images": {
"label": "Images",
"type": "image",
"display": "output"
},
"doc": {
"label": "Doc",
"type": "string",
"display": "output"
}
},
"input_names": [
"image",
"annotation_task",
"annotation_prompt",
"annotation_output_type"
],
"output_names": [
"images",
"doc"
],
"block_name": "custom",
"node_type": "custom"
}
}
}