diffusion_models.spine_dataset.spatial_transformerΒΆ

Class and helper functions used for the random deformations.

Functions

gauss_gen3D([n, s, sigma])

Generate blur kernel.

Classes

SpatialTransformer(size[, mode])

N-D Spatial Transformer.