geowatch.tasks.depth.demo_transform module

class geowatch.tasks.depth.demo_transform.Scale(size)

Bases: object

changeScale(img, size, interpolation=2)

class geowatch.tasks.depth.demo_transform.CenterCrop(size)

Bases: object

centerCrop(image, size)

class geowatch.tasks.depth.demo_transform.ToTensor

Bases: object

Convert a PIL.Image or numpy.ndarray to tensor. Converts a PIL.Image or numpy.ndarray (H x W x C) in the range [0, 255] to a torch.FloatTensor of shape (C x H x W) in the range [0.0, 1.0].

to_tensor(pic)

class geowatch.tasks.depth.demo_transform.ToNumpy

Bases: object

Converts a torch.FloatTensor of shape (C x H x W) to a numpy.ndarray (H x W x C)

to_numpy(image)

class geowatch.tasks.depth.demo_transform.Normalize(mean, std)

Bases: object

normalize(tensor, mean, std)

class geowatch.tasks.depth.demo_transform.Lighting(alphastd, eigval, eigvec)

Bases: object