configs/evaluation/projects/bridging/_preprocessing_coco_things.yaml
# @package _global_
# Preprocessing for evaluating semantic segmentation on COCO-Stuff-164k only 90 things classes.
#
# Stuff classes are set to background.
#
# Masks are in original image resolution.
dataset:
eval_transforms:
03a_preprocessing:
_target_: ocl.transforms.Map
transform:
_target_: torchvision.transforms.Compose
transforms:
- _target_: ocl.preprocessing.ConvertCocoStuff164kMasks
output_key: segmentation_mask
stuffthings_key: stuffthings_mask
ignore_key: ignore_mask
drop_stuff: true
- _target_: ocl.preprocessing.DropEntries
keys:
- stuffthings_mask
fields:
- stuffthings_mask
batch_transform: false
03b_preprocessing:
_target_: ocl.transforms.SimpleTransform
transforms:
image:
_target_: torchvision.transforms.Compose
transforms:
- _target_: torchvision.transforms.ToTensor
- _target_: torchvision.transforms.Resize
size: 224
interpolation: ${torchvision_interpolation_mode:BICUBIC}
- "${lambda_fn:'lambda image: image.clamp(0.0, 1.0)'}"
- _target_: torchvision.transforms.CenterCrop
size: 224
- _target_: torchvision.transforms.Normalize
mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225]
segmentation_mask:
_target_: torchvision.transforms.Compose
transforms:
- _target_: ocl.preprocessing.DenseMaskToTensor
ignore_mask:
_target_: torchvision.transforms.Compose
transforms:
- _target_: ocl.preprocessing.DenseMaskToTensor
batch_transform: false