`rescale_predictions=True`

1f2ec20d · srihari-humbarwadi · 19ed6e59 · 1f2ec20d
Unverified Commit 1f2ec20d authored Oct 27, 2021 by srihari-humbarwadi
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/vision/beta/projects/panoptic_maskrcnn/configs/panoptic_maskrcnn.py ...a/projects/panoptic_maskrcnn/configs/panoptic_maskrcnn.py +1 -1

No files found.
--- a/official/vision/beta/projects/panoptic_maskrcnn/configs/panoptic_maskrcnn.py
+++ b/official/vision/beta/projects/panoptic_maskrcnn/configs/panoptic_maskrcnn.py
@@ -183,7 +183,7 @@ def panoptic_fpn_coco() -> cfg.ExperimentConfig:
          model=PanopticMaskRCNN(
              num_classes=91, input_size=[1024, 1024, 3],
              panoptic_segmentation_generator=PanopticSegmentationGenerator(
-                  output_size=[640, 640]),
+                  output_size=[640, 640], rescale_predictions=True),
              stuff_classes_offset=90,
              segmentation_model=SEGMENTATION_MODEL(
                  num_classes=num_semantic_segmentation_classes,