PyTorch toolbox to work with spherical surfaces.

Note

This page is a reference documentation. It only explains the class signature, and not how to use it. Please refer to the gallery for the big picture.

class surfify.losses.vae.SphericalVAELoss(beta=9, left_mask=None, right_mask=None)[source]¶

Spherical VAE Loss.

__init__(beta=9, left_mask=None, right_mask=None)[source]¶

Init class.

Parameters

beta : float, default 9

weight of the kl divergence.

left_mask : Tensor (azimuth, elevation), default None

left cortical binary mask.

right_mask : Tensor (azimuth, elevation), default None

right cortical binary mask.