Mask

Mask(data, *[, dtype, device, requires_grad])	torch.Tensor subclass for segmentation and detection masks.
BatchMasks(data, *, samples_ranges[, dtype, ...])	torch.Tensor subclass for batch of segmentation and detection masks.

convert_batch_masks_to_masks(batch_masks)	Convert BatchMasks object to a list of Mask objects.
convert_masks_to_batch_masks(masks)	Convert a sequence of Mask objects to a BatchMasks object.