polyTEM.image.drift

DRIFT CORRECTION CODE Written by Camila Cendra, circa 2018

modified by Christina, 2023

Functions

calculate(data_tensor)
calculate_xc(tensor1, tensor2, img_size[, ...])
complex_fft(tensor[, s, normalize])	Returns powder spectra of 2D tensor (image) using PyTorch implementation.
correct_data(data, x_drift, y_drift, ...[, ...])	Corrects drift across frames in data
dose_study(data_tensor[, verbose])
hanning(N[, device])
normalize_img(img_tensor)
plot_2d_drift(x_drift, y_drift[, dx, lines, ...])	by Camila
read_mrc(fn)	Opens .mrc file containing single stack of images and returns numpy array.
tensor_shift_fft(fft)	Shift zero frequency spatial frequency component to center of 2D image.