pythonによる複数枚の画像の重ね合わせについて

Question

python初心者です。
現在、AとBのファイル内の画像を重ね合わせ
cv2.imwrite()で出力しようとしています。

具体的には
Aには、001.jpg,002.jpg・・・
Bには、001.jpg,002.jpg・・・
という画像ファイルが存在し
A内の001.jpgとB内の001.jpg、次にA内の002.jpgとB内の002.jpgとそれぞれに重ね合わせていき

それらをcv2.imwrite()で、このプログラムと同じ階層内に出力したい(重ね合わせてできた001.jpg,002.jpg・・・のような形)と考えています。

画像の処理部分はおおよそ書けましたが、出力部分がわかりません。

どなたか、助けていただけないでしょうか。

import cv2
import numpy as np
import glob

files1 =glob.glob('/A/*.jpg')
files2 =glob.glob('/B/*.jpg')

for f1 in files1:    #あとはForで1ファイルずつ実行されていく
 src = cv2.imread(f1) 
 
 for f2 in files2:
  mask = cv2.imread(f2) 
 
height, width, color = src.shape # 幅・高さ・色を取得
dst = np.zeros((height, width, 3), dtype = "uint8") # 合成画像用の変数を作成

for y in range(0, height):
   for x in range(0, width):
        if (mask[y][x] > 240).all(): 
            dst[y][x] = src[y][x] 
        else:
            dst[y][x] = 0 # 隠れている部分なので、黒にする

(追記)当然かもしれませんが、以下のようにコードを書いた際には、A内の001.jpgとB内の001.jpgしか出力されませんでした。
しかし重ね合わせ自体は成功していたので、出力部分以外には問題はないと考えています。

import cv2
import numpy as np
import glob

files1 =glob.glob('/A/*.jpg')
files2 =glob.glob('/B/*.jpg')

for f1 in files1:    #あとはForで1ファイルずつ実行されていく
 src = cv2.imread(f1) 
 
 for f2 in files2:
  mask = cv2.imread(f2) 
 
height, width, color = src.shape # 幅・高さ・色を取得
dst = np.zeros((height, width, 3), dtype = "uint8") # 合成画像用の変数を作成

for y in range(0, height):
   for x in range(0, width):
        if (mask[y][x] > 240).all(): # 「青・緑・赤」すべてが241以上なら
            dst[y][x] = src[y][x] # 隠れていない部分なので、「lena.jpg」の画素を代入
        else:
            dst[y][x] = 0 # 隠れている部分なので、黒にする
            
cv2.imwrite("dst.jpg", dst)

# cv2.imwrite("dst.jpg", dst) 
cv2.destroyAllWindows() # 終了処理

Answer 1

for f1 in files1: #あとはForで1ファイルずつ実行されていく

src = cv2.imread(f1)

こちらただfor文内で何度もイメージを変数に代入するだけ
という動作を繰り返す何ら意味がない処理になってしまっています。
この場合画像をそれぞれ代入しているsrcやmaskについては別途でリストを用意して保存すべきでしょう。

AフォルダBフォルダそれぞれ一枚ずつの画像の処理を行いたいという事であれば
for文にてzip()を使用すると良いでしょう。
また画像の保存の処理も同じく都度for文内で行う必要があります。
動作の確認は一切行っておらず、またもしかしたら質問者様の意図・目的を完全に理解しきれていない可能性がある為これで正解なのかはわかりかねますが質問者様が行いたい処理はおそらくこういう事ではないのでしょうか？

import cv2
import numpy as np
import glob

files1 =glob.glob('/A/*.jpg')
files2 =glob.glob('/B/*.jpg')


src = [cv2.imread(f1) for f1 in files1]
mask = [cv2.imread(f2) for f2 in files2]

cnt = 1
for f1, f2 in zip(src, mask):
	height, width, color = f1.shape
	dst = np.zeros((height, width, 3), dtype='uint8')
	

	# 行いたい画像の処理


	cv2.imwrite("{:03d}.jpg".format(cnt), dst)
	cnt += 1

cv2.destroyAllWindows()