Tobias Cornille commited on
Commit
391271a
1 Parent(s): 37f9596

Fix category names

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -315,6 +315,7 @@ def generate_panoptic_mask(
315
  ]
316
  stuff_category_names = [category["name"] for category in stuff_categories]
317
  thing_category_names = [category["name"] for category in thing_categories]
 
318
  else:
319
  # parse inputs
320
  thing_category_names = [
 
315
  ]
316
  stuff_category_names = [category["name"] for category in stuff_categories]
317
  thing_category_names = [category["name"] for category in thing_categories]
318
+ category_names = thing_category_names + stuff_category_names
319
  else:
320
  # parse inputs
321
  thing_category_names = [