테두리 추출용 XDoG 필터 코드 (파이썬3용)

테두리 추출용 XDoG필터 구현한 파이썬3용 코드

출처 : https://blog.csdn.net/qq_27879381/article/details/72718135

XDoG:一种快速简洁的图像边缘提取器_Python_ｉｃｍｌｌ-CSDN博客

편의를 위해 폴더 불러오는 루틴은 원본에서 일부 수정했습니다. 성능은 대단하네요. 경로명은 적당히 수정해주시기 바랍니다.

from PIL import Image,  ImageEnhance
from pylab import *
#import numpy as np
from scipy.ndimage import filters
#from skimage import io
from os import listdir
import glob, os, shutil
 
#in_dir = 'debug'
#out_dir = in_dir + '_combine'
in_dir = "d:\\input"
out_dir = "d:\\output"
#if not os.path.exists(out_dir): os.mkdir(out_dir)
 
def main():
    #for files1 in glob.glob(in_dir + './images/*.jpg'):
    for tmpfile in listdir(in_dir):
        files1 = os.path.join(in_dir, tmpfile)
 
        print("변환: ", files1)
 
 
        Gamma = 0.97
        Phi = 200
        Epsilon = 0.1
        k = 2.5
        Sigma = 1.5
 
        im = Image.open(files1).convert('L')
        im = array(ImageEnhance.Sharpness(im).enhance(3.0))
        im2 = filters.gaussian_filter(im, Sigma)
        im3 = filters.gaussian_filter(im, Sigma* k)
        differencedIm2 = im2 - (Gamma * im3)
        (x, y) = shape(im2)
        for i in range(x):
            for j in range(y):
                if differencedIm2[i, j] < Epsilon:
                    differencedIm2[i, j] = 1
                else:
                    differencedIm2[i, j] = 250 + tanh(Phi * (differencedIm2[i, j]))
 
 
        gray_pic=differencedIm2.astype(np.uint8)
        final_img = Image.fromarray( gray_pic)
        final_img.save(os.path.join(out_dir, tmpfile))
 
if __name__ == '__main__':
    main()