fixed bug ins `thing_class_ids`

7cffacfe · srihari-humbarwadi · e54c17f7 · 7cffacfe
Unverified Commit 7cffacfe authored Mar 25, 2022 by srihari-humbarwadi
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

official/vision/beta/projects/panoptic_maskrcnn/configs/panoptic_deeplab.py ...ta/projects/panoptic_maskrcnn/configs/panoptic_deeplab.py +2 -2

No files found.
--- a/official/vision/beta/projects/panoptic_maskrcnn/configs/panoptic_deeplab.py
+++ b/official/vision/beta/projects/panoptic_maskrcnn/configs/panoptic_deeplab.py
@@ -225,8 +225,8 @@ def panoptic_deeplab_coco() -> cfg.ExperimentConfig:
              post_processor=PanopticDeeplabPostProcessor(
                  output_size=input_size[:2],
                  center_score_threshold=0.1,
-                  thing_class_ids=[i for i in range(num_thing_categories)],
-                  label_divisor=256 * 256 * 256,
+                  thing_class_ids=[i for i in range(1, num_thing_categories)],
+                  label_divisor=256,
                  stuff_area_limit=4096,
                  ignore_label=ignore_label,
                  nms_kernel=41,