configs/evaluation/projects/bridging/_preprocessing_movi_c.yaml
# @package _global_
# Preprocessing for evaluating on MOVi-C
dataset:
eval_transforms:
03_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)'}" # Bicubic interpolation can get out of range
- _target_: torchvision.transforms.Normalize
mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225]
mask:
_target_: torchvision.transforms.Compose
transforms:
- _target_: ocl.preprocessing.MultiMaskToTensor
- _target_: ocl.preprocessing.CheckFormat
shape: [11, 128, 128]
one_hot: true
batch_transform: false