from PIL import Image
import matplotlib.pyplot as plt
%matplotlib inline

# 画像表示
def plt_img(img):
    plt.figure(figsize=(20,10))
    plt.imshow(img)
    plt.axis('off')
    plt.show()

# モザイク処理(mode0:モザイクをかける,mode1:モザイクを外す)
def mosaic(image,mode=0):
    pix = image.load()
    wid, hei = image.size
    s = wid * hei
    if mode==0:
        for _ in range(2):
            fl = [pix[x, y][rgb] for y in range(hei) for x in range(wid) for rgb in range(3)]
            rgb = zip(fl[:s] , fl[s:s*2] , fl[s*2:])# rgbの入れ替え
            for y in range(hei):
                for x in range(wid):
                    pix[x, y] = next(rgb)
    elif mode==1:
        for _ in range(2):
            fl = [pix[x, y][rgb] for rgb in range(3) for y in range(hei) for x in range(wid)]
            rgb = zip(fl[0::3],fl[1::3],fl[2::3])# rgbの入れ替え
            for y in range(hei):
                for x in range(wid):
                    pix[x, y] = next(rgb)

# マスク処理(mode0:マスクをかける,mode1:マスクを外す)
def mask(file_path,mode=0,m_x=(0,0),m_y=(0,0)):
    img = Image.open(file_path)
    tmp = img.copy()
    if m_x!=(0,0) and m_y!=(0,0):
        tmp = img.crop((m_x[0], m_y[0], m_x[1], m_y[1]))
    if mode==0:
        mosaic(tmp,0)
        img.paste(tmp, (m_x[0], m_y[0]))
        out_path = file_path.replace('.','_masked.'.format())
    elif mode==1:
        mosaic(tmp,1)
        img.paste(tmp, (m_x[0], m_y[0]))
        out_path = file_path.replace('_masked','_unmasked'.format())
    img.save(out_path,quality=0)
    plt_img(img)

file_path = 'ファイル名.png'
m_x = (横の始まり,横の終わり)
m_y = (縦の始まり,縦の終わり)
mask(file_path,モード(0 or 1),m_x,m_y)

タグ ：

#Python

#プログラミング

#モザイク

#画像処理