YiYiXu HF Staff commited on
Commit
d1e04bb
·
verified ·
1 Parent(s): 5b4cbe4

Update mellon_pipeline_config.json

Browse files
Files changed (1) hide show
  1. mellon_pipeline_config.json +8 -35
mellon_pipeline_config.json CHANGED
@@ -14,9 +14,12 @@
14
  "label": "Annotation Task",
15
  "type": "string",
16
  "options": [
17
- "<REFERRING_EXPRESSION_SEGMENTATION>"
 
 
 
18
  ],
19
- "value": "<REFERRING_EXPRESSION_SEGMENTATION>"
20
  },
21
  "annotation_prompt": {
22
  "label": "Annotation Prompt",
@@ -28,35 +31,12 @@
28
  "label": "Annotation Output Type",
29
  "type": "string",
30
  "options": [
31
- "mask_image"
 
 
32
  ],
33
  "value": "mask_image"
34
  },
35
- "annotation_overlay": {
36
- "label": "Annotation Overlay",
37
- "type": "boolean",
38
- "default": false
39
- },
40
- "fill": {
41
- "label": "Fill",
42
- "type": "custom",
43
- "display": "input"
44
- },
45
- "image_annotator": {
46
- "label": "Image Annotator",
47
- "type": "diffusers_auto_model",
48
- "display": "input"
49
- },
50
- "image_annotator_processor": {
51
- "label": "Image Annotator Processor",
52
- "type": "diffusers_auto_model",
53
- "display": "input"
54
- },
55
- "annotations": {
56
- "label": "Annotations",
57
- "type": "custom",
58
- "display": "output"
59
- },
60
  "images": {
61
  "label": "Images",
62
  "type": "image",
@@ -73,15 +53,8 @@
73
  "annotation_task",
74
  "annotation_prompt",
75
  "annotation_output_type",
76
- "annotation_overlay",
77
- "fill"
78
- ],
79
- "model_input_names": [
80
- "image_annotator",
81
- "image_annotator_processor"
82
  ],
83
  "output_names": [
84
- "annotations",
85
  "images",
86
  "doc"
87
  ],
 
14
  "label": "Annotation Task",
15
  "type": "string",
16
  "options": [
17
+ "<CAPTION_TO_PHRASE_GROUNDING>",
18
+ "<REFERRING_EXPRESSION_SEGMENTATION>",
19
+ "<OPEN_VOCABULARY_DETECTION>",
20
+
21
  ],
22
+ "value": "<CAPTION_TO_PHRASE_GROUNDING>"
23
  },
24
  "annotation_prompt": {
25
  "label": "Annotation Prompt",
 
31
  "label": "Annotation Output Type",
32
  "type": "string",
33
  "options": [
34
+ "bounding_box",
35
+ "mask_overlay",
36
+ "mask_image",
37
  ],
38
  "value": "mask_image"
39
  },
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
40
  "images": {
41
  "label": "Images",
42
  "type": "image",
 
53
  "annotation_task",
54
  "annotation_prompt",
55
  "annotation_output_type",
 
 
 
 
 
 
56
  ],
57
  "output_names": [
 
58
  "images",
59
  "doc"
60
  ],