Spaces:
Running
Running
Commit
·
8a8d518
1
Parent(s):
4b98e8d
Upload app.py
Browse files
app.py
CHANGED
@@ -131,11 +131,8 @@ def build_demo_classes_and_metadata(vocab, label_list):
|
|
131 |
|
132 |
MetadataCatalog.pop("fcclip_demo_metadata", None)
|
133 |
demo_metadata = MetadataCatalog.get("fcclip_demo_metadata")
|
134 |
-
demo_metadata.thing_classes =
|
135 |
-
demo_metadata.stuff_classes =
|
136 |
-
*demo_metadata.thing_classes,
|
137 |
-
*[c[0] for c in demo_stuff_classes],
|
138 |
-
]
|
139 |
demo_metadata.thing_colors = demo_thing_colors
|
140 |
demo_metadata.stuff_colors = demo_thing_colors + demo_stuff_colors
|
141 |
demo_metadata.stuff_dataset_id_to_contiguous_id = {
|
@@ -156,7 +153,6 @@ def inference(image_path, vocab, label_list):
|
|
156 |
vocab = vocab.replace(", ", ",").replace("; ", ";")
|
157 |
demo_classes, demo_metadata = build_demo_classes_and_metadata(vocab, label_list)
|
158 |
predictor.set_metadata(demo_metadata)
|
159 |
-
|
160 |
im = cv2.imread(image_path)
|
161 |
outputs = predictor(im)
|
162 |
v = OpenVocabVisualizer(im[:, :, ::-1], demo_metadata, instance_mode=ColorMode.IMAGE)
|
|
|
131 |
|
132 |
MetadataCatalog.pop("fcclip_demo_metadata", None)
|
133 |
demo_metadata = MetadataCatalog.get("fcclip_demo_metadata")
|
134 |
+
demo_metadata.thing_classes = demo_thing_classes
|
135 |
+
demo_metadata.stuff_classes = demo_thing_classes+demo_stuff_classes
|
|
|
|
|
|
|
136 |
demo_metadata.thing_colors = demo_thing_colors
|
137 |
demo_metadata.stuff_colors = demo_thing_colors + demo_stuff_colors
|
138 |
demo_metadata.stuff_dataset_id_to_contiguous_id = {
|
|
|
153 |
vocab = vocab.replace(", ", ",").replace("; ", ";")
|
154 |
demo_classes, demo_metadata = build_demo_classes_and_metadata(vocab, label_list)
|
155 |
predictor.set_metadata(demo_metadata)
|
|
|
156 |
im = cv2.imread(image_path)
|
157 |
outputs = predictor(im)
|
158 |
v = OpenVocabVisualizer(im[:, :, ::-1], demo_metadata, instance_mode=ColorMode.IMAGE)
|