reg23.ops Namespace Reference

Functions
torch.Tensor	radon2d (torch.Tensor image, torch.Tensor image_spacing, torch.Tensor phi_values, torch.Tensor r_values, int samples_per_line)
torch.Tensor	radon2d_v2 (torch.Tensor image, torch.Tensor image_spacing, torch.Tensor phi_values, torch.Tensor r_values, int samples_per_line)
torch.Tensor	d_radon2d_dr (torch.Tensor image, torch.Tensor image_spacing, torch.Tensor phi_values, torch.Tensor r_values, int samples_per_line)
torch.Tensor	radon3d (torch.Tensor volume, torch.Tensor volume_spacing, torch.Tensor phi_values, torch.Tensor theta_values, torch.Tensor r_values, int samples_per_direction)
torch.Tensor	radon3d_v2 (torch.Tensor volume, torch.Tensor volume_spacing, torch.Tensor phi_values, torch.Tensor theta_values, torch.Tensor r_values, int samples_per_direction)
torch.Tensor	d_radon3d_dr (torch.Tensor volume, torch.Tensor volume_spacing, torch.Tensor phi_values, torch.Tensor theta_values, torch.Tensor r_values, int samples_per_direction)
torch.Tensor	d_radon3d_dr_v2 (torch.Tensor volume, torch.Tensor volume_spacing, torch.Tensor phi_values, torch.Tensor theta_values, torch.Tensor r_values, int samples_per_direction)
torch.Tensor	resample_sinogram3d (torch.Tensor sinogram3d, str sinogram_type, float r_spacing, torch.Tensor projection_matrix, torch.Tensor phi_values, torch.Tensor r_values, torch.Tensor|None out=None)
torch.Tensor	resample_sinogram3d_cuda_texture (structs.CUDATexture3D texture, str sinogram_type, float r_spacing, torch.Tensor projection_matrix, torch.Tensor phi_values, torch.Tensor r_values, torch.Tensor|None out=None)
torch.Tensor	normalised_cross_correlation (torch.Tensor a, torch.Tensor b)
torch.Tensor	grid_sample3d (torch.Tensor input_, torch.Tensor grid, str address_mode_x="zero", str address_mode_y="zero", str address_mode_z="zero", torch.Tensor|None out=None)
torch.Tensor	project_drr (torch.Tensor volume, torch.Tensor voxel_spacing, torch.Tensor homography_matrix_inverse, float source_distance, int output_width, int output_height, torch.Tensor output_offset, torch.Tensor detector_spacing)
torch.Tensor	project_drr_backward (torch.Tensor volume, torch.Tensor voxel_spacing, torch.Tensor homography_matrix_inverse, float source_distance, int output_width, int output_height, torch.Tensor output_offset, torch.Tensor detector_spacing, torch.Tensor d_loss_d_drr)
torch.Tensor	project_drr_cuboid_mask (torch.Tensor volume_size, torch.Tensor voxel_spacing, torch.Tensor homography_matrix_inverse, float source_distance, int output_width, int output_height, torch.Tensor output_offset, torch.Tensor detector_spacing)
torch.Tensor	project_drrs_batched (torch.Tensor volume, torch.Tensor voxel_spacing, torch.Tensor inverse_h_matrices, float source_distance, int output_width, int output_height, torch.Tensor output_offset, torch.Tensor detector_spacing)

Function Documentation

d_radon2d_dr()

torch.Tensor reg23.ops.d_radon2d_dr	(	torch.Tensor	image,
		torch.Tensor	image_spacing,
		torch.Tensor	phi_values,
		torch.Tensor	r_values,
		int	samples_per_line
	)

d_radon3d_dr()

torch.Tensor reg23.ops.d_radon3d_dr	(	torch.Tensor	volume,
		torch.Tensor	volume_spacing,
		torch.Tensor	phi_values,
		torch.Tensor	theta_values,
		torch.Tensor	r_values,
		int	samples_per_direction
	)

d_radon3d_dr_v2()

torch.Tensor reg23.ops.d_radon3d_dr_v2	(	torch.Tensor	volume,
		torch.Tensor	volume_spacing,
		torch.Tensor	phi_values,
		torch.Tensor	theta_values,
		torch.Tensor	r_values,
		int	samples_per_direction
	)

grid_sample3d()

torch.Tensor reg23.ops.grid_sample3d	(	torch.Tensor	input_,
		torch.Tensor	grid,
		str	address_mode_x = "zero",
		str	address_mode_y = "zero",
		str	address_mode_z = "zero",
		torch.Tensor | None	out = None
	)

normalised_cross_correlation()

torch.Tensor reg23.ops.normalised_cross_correlation	(	torch.Tensor	a,
		torch.Tensor	b
	)

project_drr()

torch.Tensor reg23.ops.project_drr	(	torch.Tensor	volume,
		torch.Tensor	voxel_spacing,
		torch.Tensor	homography_matrix_inverse,
		float	source_distance,
		int	output_width,
		int	output_height,
		torch.Tensor	output_offset,
		torch.Tensor	detector_spacing
	)

project_drr_backward()

torch.Tensor reg23.ops.project_drr_backward	(	torch.Tensor	volume,
		torch.Tensor	voxel_spacing,
		torch.Tensor	homography_matrix_inverse,
		float	source_distance,
		int	output_width,
		int	output_height,
		torch.Tensor	output_offset,
		torch.Tensor	detector_spacing,
		torch.Tensor	d_loss_d_drr
	)

project_drr_cuboid_mask()

torch.Tensor reg23.ops.project_drr_cuboid_mask	(	torch.Tensor	volume_size,
		torch.Tensor	voxel_spacing,
		torch.Tensor	homography_matrix_inverse,
		float	source_distance,
		int	output_width,
		int	output_height,
		torch.Tensor	output_offset,
		torch.Tensor	detector_spacing
	)

project_drrs_batched()

torch.Tensor reg23.ops.project_drrs_batched	(	torch.Tensor	volume,
		torch.Tensor	voxel_spacing,
		torch.Tensor	inverse_h_matrices,
		float	source_distance,
		int	output_width,
		int	output_height,
		torch.Tensor	output_offset,
		torch.Tensor	detector_spacing
	)

radon2d()

torch.Tensor reg23.ops.radon2d	(	torch.Tensor	image,
		torch.Tensor	image_spacing,
		torch.Tensor	phi_values,
		torch.Tensor	r_values,
		int	samples_per_line
	)

radon2d_v2()

torch.Tensor reg23.ops.radon2d_v2	(	torch.Tensor	image,
		torch.Tensor	image_spacing,
		torch.Tensor	phi_values,
		torch.Tensor	r_values,
		int	samples_per_line
	)

Note: CPU implementation is identical to that of `radon2d`

radon3d()

torch.Tensor reg23.ops.radon3d	(	torch.Tensor	volume,
		torch.Tensor	volume_spacing,
		torch.Tensor	phi_values,
		torch.Tensor	theta_values,
		torch.Tensor	r_values,
		int	samples_per_direction
	)

radon3d_v2()

torch.Tensor reg23.ops.radon3d_v2	(	torch.Tensor	volume,
		torch.Tensor	volume_spacing,
		torch.Tensor	phi_values,
		torch.Tensor	theta_values,
		torch.Tensor	r_values,
		int	samples_per_direction
	)

Note: CPU implementation is identical to that of `radon3d`

resample_sinogram3d()

torch.Tensor reg23.ops.resample_sinogram3d	(	torch.Tensor	sinogram3d,
		str	sinogram_type,
		float	r_spacing,
		torch.Tensor	projection_matrix,
		torch.Tensor	phi_values,
		torch.Tensor	r_values,
		torch.Tensor | None	out = None
	)

resample_sinogram3d_cuda_texture()

torch.Tensor reg23.ops.resample_sinogram3d_cuda_texture	(	structs.CUDATexture3D	texture,
		str	sinogram_type,
		float	r_spacing,
		torch.Tensor	projection_matrix,
		torch.Tensor	phi_values,
		torch.Tensor	r_values,
		torch.Tensor | None	out = None
	)