【Pillow（PIL）】iPhoneのスクリーンショットをトリミングしてサイズ変更[Python]

2021年8月6日2023年7月7日

URLをコピーしました！

Pillow（PIL）

前回、Pythonの画像処理ライブラリPillow（PIL）を使って、iPhoneのスクリーンショットをトリミングしました。

プログラム全体

＜セル１＞

import os
from PIL import Image

newname = "test_"

mobile_fix_size = [0, 100, 0, 0, 1000, 0] #[left, upper, right, lower, width, height] width&height 0 = keep ratio

default_path =""

mobile_ss_size = [1125, 2436]

ext_list = [".png", ".PNG"]

＜セル２＞

def path_set(default_path):
    if default_path == "":
        path_temp = os.getcwd()
    else:
        path_temp = default_path
    return path_temp

def file_select(default_path, newname, ext_list):
    list_temp = []
    for file in os.listdir(default_path):
        if file[-4:] in ext_list:
            if not newname in file:
                list_temp.append(file)
    return list_temp
            
def trimming(im, fix_size):
    width = im.size[0]
    height = im.size[1]
    im_trim = im.crop((fix_size[0], fix_size[1], width - fix_size[2], height - fix_size[3]))
    return im_trim

def resizing(im, fix_size):
    width = im.size[0]
    height = im.size[1]
    resize_width = fix_size[4]
    resize_height = fix_size[5]
    
    if resize_width == 0 and resize_height == 0:
        print("ERROR: resize_size should not be [0, 0]")
    elif resize_height == 0:
        resized = im.resize((resize_width, int(height/(width/resize_width))))
    elif resize_width == 0:
        resized = im.resize((int(width/(height/resize_height)), resize_height))
    elif resize_size[0] != 0 and resize_size[1] != 0:
        resized = im.resize((resize_width, resize_height))
        
    return resized

＜セル３＞

if __name__ == '__main__':
    default_path = path_set(default_path)
    file_list = sorted(file_select(default_path, newname, ext_list))

    os.chdir(default_path)

    i = 1
    filling = len(str(len(file_list))) + 1
    
    for file in file_list:
        im = Image.open(file)
        if im.size[0] == mobile_ss_size[0] and im.size[1] == mobile_ss_size[1]:
            im_trim = trimming(im, mobile_fix_size)
            im_trim_resize = resizing(im_trim, mobile_fix_size)
            im_trim_resize.save(newname + str(i).zfill(filling) + ".png")
        i = i + 1

サイズ変更のための関数

今回まず追加したのはサイズ変更のための関数です。

つまり＜セル２＞のこの部分。

def resizing(im, fix_size):
    width = im.size[0]
    height = im.size[1]
    resize_width = fix_size[4]
    resize_height = fix_size[5]
    
    if resize_width == 0 and resize_height == 0:
        print("ERROR: resize_size should not be [0, 0]")
    elif resize_height == 0:
        resized = im.resize((resize_width, int(height/(width/resize_width))))
    elif resize_width == 0:
        resized = im.resize((int(width/(height/resize_height)), resize_height))
    elif resize_size[0] != 0 and resize_size[1] != 0:
        resized = im.resize((resize_width, resize_height))
        
    return resized

この関数ではpillowで読み込んだ画像imと変更後の画像サイズの定義であるmobile_fix_size（関数ないでは一般化してあるためfix_size）を使います。

まず「im.size[0]」と「im.size[1]」を使って、画像の幅を変数widthに、高さを変数heightに代入しておきます。

またサイズ変更の幅と高さのサイズはmobile_fix_sizeの5番目と6番目の数字として定義してあるためその値をresize_widthとresize_heightに代入しておきます。

resize_widthとresize_heightですが、例えばresize_widthが０なら縦横比をそのままにresize_heightに高さを合わせ、resize_heightが０なら縦横比をそのままにresize_widhtに幅を合わせるようにします。

そのため両方とも０の場合は、エラーを表示するということが必要になります。

また両方とも値が入っている場合は、縦横比を気にせずに幅、高さをそれぞれの値にする。

つまり4つの場合分けが必要になります。

その部分がこちらです。

    if resize_width == 0 and resize_height == 0:
        print("ERROR: resize_size should not be [0, 0]")
    elif resize_height == 0:
        resized = im.resize((resize_width, int(height/(width/resize_width))))
    elif resize_width == 0:
        resized = im.resize((int(width/(height/resize_height)), resize_height))
    elif resize_size[0] != 0 and resize_size[1] != 0:
        resized = im.resize((resize_width, resize_height))

上から両方とも０だった場合、resize_heightが０だった場合、resize_widthが０だった場合、両方とも値が入っていた場合です。

ここでresize_heightが０だった場合、resize_widthが０だった場合に関しては前に作成した「縦横比を変える、変えないどちらにでも対応できるようにする」プログラムを改変して用います。