MCPcopy Create free account
hub / github.com/NVIDIA/semantic-segmentation / MaskToTensor

Class MaskToTensor

transforms/transforms.py:70–72  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

68
69
70class MaskToTensor(object):
71 def __call__(self, img, blockout_predefined_area=False):
72 return torch.from_numpy(np.array(img, dtype=np.int32)).long()
73
74class RelaxedBoundaryLossToTensor(object):
75 """

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected