Skip to content

GitLab

Commit 96e9df0a authored by Abdullah Rashwan's avatar Abdullah Rashwan Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 339598839
parent f1651d14
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
official/vision/beta/configs/semantic_segmentation.py
View file @ 96e9df0a
...@@ -39,6 +39,7 @@ class DataConfig(cfg.DataConfig): ...@@ -39,6 +39,7 @@ class DataConfig(cfg.DataConfig):
groundtruth_padded_size: List[int] = dataclasses.field(default_factory=list) groundtruth_padded_size: List[int] = dataclasses.field(default_factory=list)
aug_scale_min: float = 1.0 aug_scale_min: float = 1.0
aug_scale_max: float = 1.0 aug_scale_max: float = 1.0
drop_remainder: bool = True
@dataclasses.dataclass @dataclasses.dataclass
...@@ -138,7 +139,8 @@ def seg_deeplabv3_pascal() -> cfg.ExperimentConfig: ...@@ -138,7 +139,8 @@ def seg_deeplabv3_pascal() -> cfg.ExperimentConfig:
is_training=False, is_training=False,
global_batch_size=eval_batch_size, global_batch_size=eval_batch_size,
resize_eval_groundtruth=False, resize_eval_groundtruth=False,
groundtruth_padded_size=[512, 512]), groundtruth_padded_size=[512, 512],
drop_remainder=False),
# resnet50 # resnet50
init_checkpoint='gs://cloud-tpu-checkpoints/vision-2.0/deeplab/deeplab_resnet50_imagenet/ckpt-62400', init_checkpoint='gs://cloud-tpu-checkpoints/vision-2.0/deeplab/deeplab_resnet50_imagenet/ckpt-62400',
init_checkpoint_modules='backbone'), init_checkpoint_modules='backbone'),
......
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