Skip to content

GitLab

Unverified Commit 1f2ec20d authored by srihari-humbarwadi's avatar srihari-humbarwadi
Browse files

`rescale_predictions=True`

parent 19ed6e59
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/vision/beta/projects/panoptic_maskrcnn/configs/panoptic_maskrcnn.py
View file @ 1f2ec20d
...@@ -183,7 +183,7 @@ def panoptic_fpn_coco() -> cfg.ExperimentConfig: ...@@ -183,7 +183,7 @@ def panoptic_fpn_coco() -> cfg.ExperimentConfig:
model=PanopticMaskRCNN( model=PanopticMaskRCNN(
num_classes=91, input_size=[1024, 1024, 3], num_classes=91, input_size=[1024, 1024, 3],
panoptic_segmentation_generator=PanopticSegmentationGenerator( panoptic_segmentation_generator=PanopticSegmentationGenerator(
output_size=[640, 640]), output_size=[640, 640], rescale_predictions=True),
stuff_classes_offset=90, stuff_classes_offset=90,
segmentation_model=SEGMENTATION_MODEL( segmentation_model=SEGMENTATION_MODEL(
num_classes=num_semantic_segmentation_classes, num_classes=num_semantic_segmentation_classes,
......
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