フォルダ内の画像全てを加工して，加工前の名前で保存したい

Question

現在，以下のように画像にマスキングを施しているのですが，これをフォルダ内の画像全てに適応して，加工前の名前（このコード上では「2019」など）で保存したいです．一応，globを使ったコードを書いてみたのですが，書いた所から先が，どのようにすれば良いか分かりません．ご指摘よろしくお願いします．
一つ目のコードは動作確認済です．二つ目のコードをRUNしても，エラーも出ず，保存先に画像も保存されておりません．

# -*- coding: utf-8 -*-

import cv2
import numpy as np

def main():
    
    img = cv2.imread("C:/data/2019.jpeg")

    mask = cv2.imread("C:/data/1.jpg", 0)

    img2 = cv2.bitwise_and(img, img, mask=mask)

    cv2.imwrite(C:/data/masked/2019.jpg", img2)


if __name__ == '__main__':
    main()

以下に，加筆したものを記します．

# -*- coding: utf-8 -*-

import cv2
import numpy as np
import glob


files = glob.glob("C:/data/folder/*.jpg")
for file in files:


    def main():

        img = cv2.imread(file)

        mask = cv2.imread("C:/data/1.jpg", 0)

        img2 = cv2.bitwise_and(img, img, mask=mask)

        cv2.imwrite("C:/masked/*.jpg", img2)


    if __name__ == '__main__':
        main()

Answer 1

cv2.imwrite("C:/masked/*.jpg", img2)
のようにしても * の部分に対応するファイル名を補完してくれたりはしませんよ。
出力するファイル名は自分で決める必要があります。

os モジュールを使って、もとのファイル名を取得して以下のようにしてください。

※ このままでは動かないので、適宜変えてください。

import glob
import os

import cv2
import numpy as np

out_dir = 'output'
os.makedirs(out_dir, exist_ok=True)

for path in glob.glob("C:/data/folder/*.jpg"):
    src = cv2.imread(path)

    # なにか処理

    # 出力パスを決める
    basename = os.path.basename(path)
    out_path = os.path.join(out_dir, basename)
    # 保存
    cv2.imwrite(out_path, dst)

Comments

[blackmk]

ありがとうございます。スムーズに動きました！

Answer 2

基本的に、関数はトップレベルで定義するようにした方がコードは書きやすいです。

import glob
import cv2


_mask = cv2.imread("C:/data/1.jpg", 0)
assert _mask is not None

def en_mask(p):
    img = cv2.imread(p)
    assert img is not None

    return cv2.bitwise_and(img, img, mask=_mask)


def main():
    for path in glob.glob("C:/data/folder/*.jpg"):
        masked = en_mask(path)
        
        name = path.split('\')[-1]
        cv2.imwrite(f'C:/masked/{name}', masked)


if __name__ == '__main__':
    main()

動作未検証ですので、ちょっとしたバグを埋め込んでいるかもしれません。悪しからず。

Comments

[blackmk]

ありがとうございます！スムーズに動きました！コードのご指摘もありがとうございます．