diff --git a/datasets/raise/fft/generate_vignetting.py b/datasets/raise/fft/generate_vignetting.py new file mode 100644 index 0000000..0a4d16e --- /dev/null +++ b/datasets/raise/fft/generate_vignetting.py @@ -0,0 +1,28 @@ +import numpy as np +import matplotlib.pyplot as plt + +A = -1.076923838846692e-06 +B = 0.0010539307456559996 +C = 31.8 + +SHAPE = (2_000, 3_000) +CENTER = np.array(SHAPE) // 2 + +# `dtype` by default is `float64` as wanted. +image = np.empty(SHAPE) + +def getDistance(y, x): + d = ((y - (SHAPE[0] / 2)) ** 2 + (x - (SHAPE[1] / 2)) ** 2) ** 0.5 + return d + +def vignetting(y, x): + d = getDistance(y, x) + return A * (d ** 2) + B * (d ** 2) + C + +for y in range(SHAPE[0]): + for x in range(SHAPE[1]): + image[y, x] = vignetting(y, x) + +print(image[np.array(SHAPE) / 2]) +plt.imshow(image) +plt.show() \ No newline at end of file