Skip to content

configs/evaluation/projects/bridging/metrics_movi_e_sa.yaml

# @package _global_
# Evaluate on MOVi-E with slot attention

defaults:
  - /evaluation_config  # (1)!
  - /evaluation/projects/bridging/_base_metrics # (2)!
  - /evaluation/projects/bridging/_preprocessing_movi_e # (3)!
  - /evaluation/projects/bridging/_metrics_discovery_movi # (4)!
  - /dataset: movi_e_image  # (5)!
  - _self_

eval_batch_size: 16

modules:
  masks_resized:
    size_tensor_path: input.mask
    patch_mode: false

plugins:
  03_preprocessing:
    evaluation_transforms:
      image:
        _target_: torchvision.transforms.Compose
        transforms:
          - _target_: torchvision.transforms.ToTensor
          - _target_: torchvision.transforms.Normalize
            mean: [0.5, 0.5, 0.5]
            std: [0.5, 0.5, 0.5]
  1. /evaluation_config
  2. /evaluation/projects/bridging/_base_metrics
  3. /evaluation/projects/bridging/_preprocessing_movi_e
  4. /evaluation/projects/bridging/_metrics_discovery_movi
  5. /dataset/movi_e_image