Pytorch Module

The pytorch module contains functions for handling computed tomography and chemical imaging data using Pytorch.

• Models Module
  • CNN1D
  • CNN2D
  • CNN3D
  • ConvBlock
  • Crop2D
  • DownBlock
  • PeakFitCNN
  • PeakModel
  • PrmCNN2D
  • ResNet2D
  • ResNetBlock
  • SD2I
  • UNet2D
  • UpBlock
  • VolumeModel
• Peak Models Module
  • chebyshev_polynomial()
  • doniach_sunjic_peak()
  • exponential_power_peak()
  • gaussian_peak()
  • lorentzian_peak()
  • pearson7_peak()
  • polynomial()
  • pseudo_voigt_peak()
  • pseudo_voigt_peak_t()
  • split_pearson_peak()
  • voigt_peak()
• Tomography Module
  • Amatrix_rec()
  • Amatrix_sino()
  • Amatrix_torch()
  • RotMat()
  • Sino_torch()
  • back_project_3D()
  • cgls_pytorch_functional()
  • compute_W_ray()
  • create_torch_Amat()
  • forward_project_3D()
  • imrotate_torch()
  • sirt_pytorch_functional()
• Utilities Module
  • SSIM2D
  • SSIM3DLoss
  • calc_patches_indices()
  • denormalize()
  • draw_valid_indices()
  • filter_patch_indices()
  • generate_indices()
  • generate_sobol_indices_batch()
  • initialize_counter()
  • tv_3d()
  • tv_3d_spectral()
  • tv_spatial()
  • tv_spectral()
  • update_counter()
• Registration Module
  • NCC
  • build_affine_matrix()
  • build_affine_matrix_3d()
  • icp_torch()
  • mae_loss()
  • mse_loss()
  • normalize_affine_matrix()
  • register_affine_2d()
  • register_affine_3d()
  • to_tensor2d()
  • to_tensor3d()
  • warp_volume_xy_batched()