osc.data.clevr_with_masks.prepare_test_segmentation

prepare_test_segmentation(example, img_size: Tuple[int, int], crop_size: Tuple[int, int], mean: Tuple[float, float, float], std: Tuple[float, float, float])

Prepare a test example for segmentation (center crop+normalization)

Parameters

• example –

• img_size (Tuple[int, int]) – image size (H, W)

• crop_size (Tuple[int, int]) – crop size (H, W)

• mean (Tuple[float, float, float]) – image mean for normalization

• std (Tuple[float, float, float]) – image standard deviation for normalization

Returns

A dict containing the image [3 H W], the mask [C H W] and a bool vector of object visibility [C]