Add alpha channel to image where background pixels are transparent

Load configs, initialize CUDA, CuDNN and the random seeds.

Convert Omega Conf or dataclass to python dict recursively for wandb logging

register hook to extract features. :param layers: layers from which to extract features. :param facet: facet to extract. One

unregisters the hooks. should be called after feature extraction.

Search the specified/latest checkpoint in checkpoint_dir and load the model and optimizer.

Transform points. Args: points: Tensor containing 3D points with shape [minibatch_size, num_vertices, 3] or [1, num_vertices, 3] m

[Borrowed from PyTorch3D] Return the rotation matrices for one of the rotations about an axis of which Euler angles describe, for each value

(0, 0, 1) = matmul(b, R^(-1)) assumes y, z is a symmetry plane returns R