【Python】フォルダ内の画像を一度にサイズ変更する

#coding: utf-8
import cv2
import os
import sys
import shutil
import unicodedata# パラメーター
WIDTH = 800
HEIGHT = None
SAVE_PATH = “resized_image”

# ファイルの拡張子を調べる関数
def check_img(file_name):
extention = file_name[-3:]

if extention in {“jpg”, “JPG”, “png”, “PNG”}:
result = True
else:
result = False

return result

# リサイズするサイズを算出する関数
# widthかheightのどちらかが指定されていれば、
# アスペクト比を維持したままサイズを変更する。
# 両方指定されていれば、そのサイズにリサイズする
def decide_size(img, width=None, height=None):
if width is not None and height is None:
height = int(img.shape[0] * (width / img.shape[1]))
elif width is None and height is not None:
width = int(img.shape[1] * (height / img.shape[0]))

return (width, height)

# ファイル名に日本語が含まれているかチェックする関数
def is_japanese(file_name):
for ch in file_name:
name = unicodedata.name(ch)
if “CJK UNIFIED” in name \
or “HIRAGANA” in name \
or “KATAKANA” in name:
return True
return False

if __name__ == “__main__”:

# 引数の取得
args = sys.argv

# 引数によって読み込み先を変える
if len(args) >= 2:
load_path = args[1]
else:
load_path = os.getcwd()

# カレントディレクトリを移動する
os.chdir(load_path)
cd = os.getcwd()
print(“Change path here : {}” .format(cd))

# 読み込み先のディレクトリ内のファイルを取得
img_list = os.listdir(cd)

# 保存用のフォルダを作成
save_path = os.path.join(“.”, SAVE_PATH)
if os.path.exists(save_path):
pass
else:
os.mkdir(save_path)
print(“SAVE_PATH was just made now : “, save_path)

# 画像ファイルのみをリサイズして保存
for num, img_name in enumerate(img_list):

# 画像の拡張子をチェックする
if check_img(img_name):
print(“—– {} —–” .format(img_name))

# ファイル名に日本語が含まれているかチェックする
if is_japanese(img_name):
save_name = os.path.join(save_path, “resized_image_” + str(num) + “.jpg”)
shutil.copy(img_name, save_name)
img_name = save_name
print(“Renamed : {}” .format(save_name))
else:
save_name = os.path.join(save_path, img_name)

# 画像を読み込む
img = cv2.imread(img_name)

# サイズを決めてリサイズする
size = decide_size(img, WIDTH, HEIGHT)
new_img = cv2.resize(img, dsize=size)

# リサイズ後の画像を保存する
if cv2.imwrite(save_name, new_img):
print(“Succeced to save.”)
else:
print(“Faild to save.”)

#上記ソースコード：10，11行目
WIDTH = 800
HEIGHT = None#上記ソースコード：31～37行目
def decide_size(img, width=None, height=None):
if width is not None and height is None:
height = int(img.shape[0] * (width / img.shape[1]))
elif width is None and height is not None:
width = int(img.shape[1] * (height / img.shape[0]))

return (width, height)

#上記ソースコード：98，99行目
size = decide_size(img, WIDTH, HEIGHT)
new_img = cv2.resize(img, dsize=size)

#上記ソースコード：57～60行目
if len(args) >= 2:
load_path = args[1]
else:
load_path = os.getcwd()

#上記ソースコード：16～24行目
def check_img(file_name):
extention = file_name[-3:]if extention in {“jpg”, “JPG”, “png”, “PNG”}:
result = True
else:
result = False

return result

#上記ソースコード：68行目
img_list = os.listdir(cd)

#上記ソースコード：79行目～
for num, img_name in enumerate(img_list):
if check_img(img_name):
（for文内は中略）

# 上記ソースコード：41～48行目
def is_japanese(file_name):
for ch in file_name:
name = unicodedata.name(ch)
if “CJK UNIFIED” in name \
or “HIRAGANA” in name \
or “KATAKANA” in name:
return True
return False# 上記ソースコード：86～92行目
if is_japanese(img_name):
save_name = os.path.join(save_path, “resized_image_” + str(num) + “.jpg”)
shutil.copy(img_name, save_name)
img_name = save_name
print(“Renamed : {}” .format(save_name))
else:
save_name = os.path.join(save_path, img_name)