Skip to content

GitLab

Commit 3570bd34 authored by Abdullah Rashwan's avatar Abdullah Rashwan Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 362200746
parent 766e019a
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
official/vision/beta/configs/semantic_segmentation.py
View file @ 3570bd34
...@@ -405,7 +405,9 @@ def seg_deeplabv3plus_cityscapes() -> cfg.ExperimentConfig: ...@@ -405,7 +405,9 @@ def seg_deeplabv3plus_cityscapes() -> cfg.ExperimentConfig:
config = cfg.ExperimentConfig( config = cfg.ExperimentConfig(
task=SemanticSegmentationTask( task=SemanticSegmentationTask(
model=SemanticSegmentationModel( model=SemanticSegmentationModel(
num_classes=20, # Cityscapes uses only 19 semantic classes for train/evaluation.
# The void (background) class is ignored in train and evaluation.
num_classes=19,
input_size=[None, None, 3], input_size=[None, None, 3],
backbone=backbones.Backbone( backbone=backbones.Backbone(
type='dilated_resnet', dilated_resnet=backbones.DilatedResNet( type='dilated_resnet', dilated_resnet=backbones.DilatedResNet(
......
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