MCPcopy Create free account
hub / github.com/Engineering-Course/LIP_JPPNet / prepare_label

Function prepare_label

utils/utils.py:47–63  ·  view source on GitHub ↗

Resize masks and perform one-hot encoding. Args: input_batch: input tensor of shape [batch_size H W 1]. new_size: a tensor with new height and width. Returns: Outputs a tensor of shape [batch_size h w 21] with last dimension comprised of 0's and 1's only.

(input_batch, new_size, one_hot=True)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

mainFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected