this repo has no description
dotfiles
at main 44 lines 1.1 kB view raw
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)