this repo has no description
dotfiles
1#!/usr/bin/env python3
2
3# pylint: disable=C0103
4
5import sys
6from scipy import absolute, fftpack
7import skimage
8import imageio
9
10scale = 5.
11
12image = imageio.imread(sys.argv[1])
13image = skimage.transform.rescale(
14 skimage.color.rgba2rgb(image), 1./scale, anti_aliasing=False, multichannel=True
15 )
16fft = fftpack.fftn(image)
17fshift = fftpack.fftshift(fft)
18
19rows = len(fft)
20cols = len(fft[0])
21red_const = int(3*cols/4)
22
23crow = rows/2
24ccol = cols/2
25
26upper_boundary = int(crow - rows / red_const)
27lower_boundary = int(crow + rows / red_const)
28left_boundary = int(ccol - cols / red_const)
29right_boundary = int(ccol + cols / red_const)
30
31fshift[:upper_boundary, :left_boundary] = 0
32fshift[:upper_boundary, right_boundary:] = 0
33fshift[lower_boundary:, right_boundary:] = 0
34fshift[lower_boundary:, :left_boundary] = 0
35
36f_ishift = fftpack.ifftshift(fshift)
37img_back = fftpack.ifftn(f_ishift)
38img_back = absolute(img_back)
39
40img_back = skimage.transform.rescale(
41 img_back, scale, anti_aliasing=False, multichannel=True, order=0
42 )
43
44result = imageio.imwrite(sys.argv[2], img_back)