MCPcopy Create free account
hub / github.com/apple/ml-4m / __init__

Method __init__

fourm/data/modality_transforms.py:572–574  ·  view source on GitHub ↗
(self, mask_pool_size=1)

Source from the content-addressed store, hash-verified

570class MaskTransform(ImageTransform):
571
572 def __init__(self, mask_pool_size=1):
573 assert isinstance(mask_pool_size, int)
574 self.mask_pool_size = mask_pool_size # Use to expand masks
575
576 def mask_to_tensor(self, img):
577 mask = TF.to_tensor(img)

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected