Spaces:

segments
/

panoptic-segment-anything

Runtime error

Tobias Cornille commited on Apr 18, 2023

Commit

938273e

1 Parent(s): 8c3fbdd

Fix bitmap file output

Files changed (1) hide show

app.py CHANGED Viewed

@@ -407,7 +407,7 @@ def generate_panoptic_mask(
         for i, panoptic_name in enumerate(panoptic_names)
     ]
-    output_file_path = "tmp/segmentation_bitmap.png"
     stuff_category_ids = [category_name_to_id[name] for name in stuff_category_names]
     annotations = inds_to_segments_format(
         panoptic_inds, thing_category_ids, stuff_category_ids, output_file_path

         for i, panoptic_name in enumerate(panoptic_names)
     ]
+    output_file_path = "output_segmentation_bitmap.png"
     stuff_category_ids = [category_name_to_id[name] for name in stuff_category_names]
     annotations = inds_to_segments_format(
         panoptic_inds, thing_category_ids, stuff_category_ids, output_file_path