diffusion_models.mri_forward.undersampling_mask

Functions

gaussian_kernel1d_h(size, rel_sigma)	1D Gaussian kernel repeated along horizontal dimension.
gaussian_kernel1d_v(size, rel_sigma)	1D Gaussian kernel repeated along vertical dimension.
gaussian_kernel2d(size, rel_sigma)	Gaussian kernel as 2D tensor.
naive_undersampling1d_h(size, ...)	Regular subsampling of horizontal lines with given undersampling ratio.
naive_undersampling1d_v(size, ...)	Regular subsampling of vertical lines with given undersampling ratio.
naive_undersampling2d(size, undersampling_ratio)	2D regular subsampling with given undersampling ratio.
normalize_mask(mask)

Classes

StochasticUndersamplingMask(mask_type, ...)
UndersamplingMask(mask_type, undersampling_ratio)