vis4d.data.transforms.to_tensor¶

ToTensor transformation.

Classes

ToTensor(*[, in_keys, out_keys, sensors, ...])

Transform all entries in a list of DataDict from numpy to torch.

class ToTensor(*, in_keys=['data'], out_keys=['data'], sensors=None, same_on_batch=True, **kwargs)[source]¶

Transform all entries in a list of DataDict from numpy to torch.

Note that we reshape K.images from NHWC to NCHW.

__call__(batch)[source]¶

Transform all entries to tensor.