patch_denoise.gpu.dataloader#

Torch dataloader for the noisy data.

Functions

patchify_tensor

Transform a tensor into a collection of patches with specified shape and overlap.

select_patches_to_process

Select patches to process based on the mask and threshold.

sliding_sum_nd

Compute a sliding sum across all dimensions using sequential 1D convolutions.

Classes

PatchDataset

GPU-resident collection of patches, batch-gathered from input_data.