Python – ディレクトリ内のファイルに同じ処理を行う方法

from pathlib import Path


def get_paths(input_dir, exts=None):
    paths = sorted([x for x in input_dir.iterdir()])
    if exts:
        paths = list(filter(lambda x: x.suffix in exts, paths))

    return paths


# ディレクトリ内の指定した拡張子のファイルをすべて取得する。
input_dir = Path(r"/data/samples")
for path in get_paths(input_dir, exts=[".jpg", ".jpeg", ".png"]):
    print(path)

import cv2

from pathlib import Path


def get_paths(input_dir, exts=None):
    paths = sorted([x for x in input_dir.iterdir()])
    if exts:
        paths = list(filter(lambda x: x.suffix in exts, paths))

    return paths


# ディレクトリ内の指定した拡張子のファイルをすべて取得する。
input_dir = Path(r"/data/samples")

# 出力先のディレクトリを作成する。
output_dir = Path(r"output")
output_dir.mkdir(exist_ok=True)

for path in get_paths(input_dir, exts=[".jpg", ".jpeg", ".png"]):
    # 画像を読み込む。
    img = cv2.imread(str(path))
    # リサイズする。
    dst = cv2.resize(img, (300, 300))
    # 結果を保存する。
    save_path = output_dir / path.name
    cv2.imwrite(str(save_path), dst)

from pathlib import Path


def get_paths(input_dir, exts=None):
    paths = sorted([x for x in input_dir.glob("**/*")])
    if exts:
        paths = list(filter(lambda x: x.suffix in exts, paths))

    return paths


# ディレクトリ内の指定した拡張子のファイルをすべて取得する。
input_dir = Path(r"sample")
for path in get_paths(input_dir, exts=[".jpg", ".jpeg", ".png"]):
    print(path)

from concurrent import futures
from pathlib import Path

import cv2


def get_paths(input_dir, exts=None):
    paths = sorted([x for x in input_dir.iterdir()])
    if exts:
        paths = list(filter(lambda x: x.suffix in exts, paths))

    return paths


# ディレクトリ内の指定した拡張子のファイルをすべて取得する。
input_dir = Path(r"/data/samples")

# 処理対象のファイル一覧を取得する。
input_paths = get_paths(input_dir, exts=[".jpg", ".jpeg", ".png"])

# 出力先のディレクトリを作成する。
output_dir = Path(r"output")
output_dir.mkdir(exist_ok=True)


def process(path):
    """処理内容を関数にまとめる
    """
    # 画像を読み込む。
    img = cv2.imread(str(path))
    # リサイズする。
    dst = cv2.resize(img, (300, 300))
    # 結果を保存する。
    save_path = output_dir / path.name

    return cv2.imwrite(str(save_path), dst)


with futures.ThreadPoolExecutor() as executor:
    rets = executor.map(process, input_paths[:10])

# 結果を受け取る。
print([x for x in rets])