Skip to content

GitLab

Unverified Commit 7cffacfe authored by srihari-humbarwadi's avatar srihari-humbarwadi
Browse files

fixed bug ins `thing_class_ids`

parent e54c17f7
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/vision/beta/projects/panoptic_maskrcnn/configs/panoptic_deeplab.py
View file @ 7cffacfe
...@@ -225,8 +225,8 @@ def panoptic_deeplab_coco() -> cfg.ExperimentConfig: ...@@ -225,8 +225,8 @@ def panoptic_deeplab_coco() -> cfg.ExperimentConfig:
post_processor=PanopticDeeplabPostProcessor( post_processor=PanopticDeeplabPostProcessor(
output_size=input_size[:2], output_size=input_size[:2],
center_score_threshold=0.1, center_score_threshold=0.1,
thing_class_ids=[i for i in range(num_thing_categories)], thing_class_ids=[i for i in range(1, num_thing_categories)],
label_divisor=256 * 256 * 256, label_divisor=256,
stuff_area_limit=4096, stuff_area_limit=4096,
ignore_label=ignore_label, ignore_label=ignore_label,
nms_kernel=41, nms_kernel=41,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment