Spaces:
Runtime error
Runtime error
Tobias Cornille
commited on
Commit
·
37f9596
1
Parent(s):
938273e
Fix task attributes json
Browse files
app.py
CHANGED
|
@@ -293,9 +293,9 @@ def generate_panoptic_mask(
|
|
| 293 |
segmentation_background_threshold=0.1,
|
| 294 |
shrink_kernel_size=20,
|
| 295 |
num_samples_factor=1000,
|
| 296 |
-
task_attributes_json=
|
| 297 |
):
|
| 298 |
-
if task_attributes_json is not
|
| 299 |
task_attributes = json.loads(task_attributes_json)
|
| 300 |
categories = task_attributes["categories"]
|
| 301 |
category_name_to_id = {
|
|
@@ -304,10 +304,14 @@ def generate_panoptic_mask(
|
|
| 304 |
# split the categories into "stuff" categories (regions w/o instances)
|
| 305 |
# and "thing" categories (objects/instances)
|
| 306 |
stuff_categories = [
|
| 307 |
-
category
|
|
|
|
|
|
|
| 308 |
]
|
| 309 |
thing_categories = [
|
| 310 |
-
category
|
|
|
|
|
|
|
| 311 |
]
|
| 312 |
stuff_category_names = [category["name"] for category in stuff_categories]
|
| 313 |
thing_category_names = [category["name"] for category in thing_categories]
|
|
|
|
| 293 |
segmentation_background_threshold=0.1,
|
| 294 |
shrink_kernel_size=20,
|
| 295 |
num_samples_factor=1000,
|
| 296 |
+
task_attributes_json="",
|
| 297 |
):
|
| 298 |
+
if task_attributes_json is not "":
|
| 299 |
task_attributes = json.loads(task_attributes_json)
|
| 300 |
categories = task_attributes["categories"]
|
| 301 |
category_name_to_id = {
|
|
|
|
| 304 |
# split the categories into "stuff" categories (regions w/o instances)
|
| 305 |
# and "thing" categories (objects/instances)
|
| 306 |
stuff_categories = [
|
| 307 |
+
category
|
| 308 |
+
for category in categories
|
| 309 |
+
if "has_instances" not in category or not category["has_instances"]
|
| 310 |
]
|
| 311 |
thing_categories = [
|
| 312 |
+
category
|
| 313 |
+
for category in categories
|
| 314 |
+
if "has_instances" in category and category["has_instances"]
|
| 315 |
]
|
| 316 |
stuff_category_names = [category["name"] for category in stuff_categories]
|
| 317 |
thing_category_names = [category["name"] for category in thing_categories]
|