def setUpClass(cls):
nx0 = 48
ny0 = 96
nx1 = 32
ny1 = 64
# Sometimes errors with nx = ny = 16
# nx = 16
# ny = 16
displacement_x = 3.3
displacement_y = 5.8
cls.displacements = np.array([displacement_x, displacement_y])
nb_particles = (max(nx0, ny0) // 4)**2
cls.im0, cls.im1 = make_synthetic_images(
cls.displacements,
nb_particles,
shape_im0=(ny0, nx0),
shape_im1=(ny1, nx1),
epsilon=0.0,
)
cls.im1 = cls.im1.astype("float32")
def setUpClass(cls):
nx = 32
ny = 32
# Sometimes errors with nx = ny = 16
# nx = 16
# ny = 16
displacement_x = 3.3
displacement_y = 5.8
cls.displacements = np.array([displacement_x, displacement_y])
nb_particles = (nx // 4)**2
cls.im0, cls.im1 = make_synthetic_images(cls.displacements,
nb_particles,
shape_im0=(ny, nx),
epsilon=0.0)
from fluidimage.synthetic import make_synthetic_images
nx = 32
ny = 32
displacement_x = 2.0
displacement_y = 2.0
displacements = np.array([displacement_x, displacement_y])
nb_particles = (nx // 3)**2
print("nx: {} ; ny: {}".format(nx, ny))
im0, im1 = make_synthetic_images(displacements,
nb_particles,
shape_im0=(ny, nx),
epsilon=0.0)
# plt.figure()
# ax0 = plt.subplot(121)
# ax1 = plt.subplot(122)
# axi0 = ax0.imshow(im0, interpolation='nearest')
# axi1 = ax1.imshow(im1, interpolation='nearest')
classes = {
"sig": CorrelScipySignal,
"ndimage": CorrelScipyNdimage,
"np.fft": CorrelFFTNumpy,
"fftw": CorrelFFTW,