to_displayable_masks

hybrid_learning.experimentation.fuzzy_exp.fuzzy_exp_vis.to_displayable_masks(masks, target_size=None)

Turn all mask items of masks into shape [h,w], potentially splitting up stacked masks.

Parameters

• masks (Dict[str, Tensor]) –

• target_size (Optional[Tuple[int, int]]) –

Return type

Dict[str, Tensor]