2つのフォルダに入った画像を順番に連結させたい

Question

下記のサイトで記載されているパターンに似ておりますが、
フォルダAとフォルダB内の画像のすべての組み合わせでなく、順番に横に連結させたいです。
https://teratail.com/questions/284453
https://teratail.com/questions/283284
https://teratail.com/questions/284253

下記コードですと、フォルダAの画像枚数×フォルダBの画像枚数のとてつもない数の新しい画像ができてしまいます。

from PIL import Image
import itertools
from pathlib import Path

def get_concat_h(im1, im2, color="black"):
    dst = Image.new(
        'RGB', (im1.width + im2.width, im1.height))
    dst.paste(im1, (0, 0))
    dst.paste(im2, (im1.width, 0))
    return dst

img_dir1 = Path("フォルダの場所/trainA")  # 左側の画像があるディレクトリ
img_dir2 = Path("フォルダの場所/trainB")  # 右側の画像があるディレクトリ
output_dir = Path("フォルダの場所/combined")  # 出力ディレクトリ
output_dir.mkdir(exist_ok=True)

for path1, path2 in itertools.product(img_dir1.iterdir(), img_dir2.iterdir()):
    print(f"concat {path1} and {path2}")
    img1 = Image.open(path1)
    img2 = Image.open(path2)

    dst = get_concat_h(img1, img2)

    save_path = output_dir / f"{path1.stem}_{path2.stem}.jpg"
    dst.save(save_path)

目的は下記フォルダCに生成された画像をpix2pixのデータセットにすることです。

フォルダAには256 x 256の画像
フォルダBには256 x 256の画像
フォルダCには上記を横に連結した512 x 256の画像

A
├── a.jpg
└── b.jpg
└── c.jpg
└── d.jpg
└── e.jpg

B
├── f.jpg
└── g.jpg
└── h.jpg
└── i.jpg
└── j.jpg

output
├── a_f.jpg
├── b_g.jpg
├── c_h.jpg
└── d_i.jpg
└── e_j.jpg

ご教授いただけますと幸いです。
よろしくお願いいたします。

Answer 1

ディレクトリAとBに入っているJPGファイルをどのように対応させたいのかが不明なので、これでOKということにはならないかもしれませんが、それぞれのディレクトリのファイルを1つずつ対応させたいのであれば、forの行をこんなふうにすればどうでしょうか。

for path1, path2  in zip(list(img_dir1.glob("*.jpg")), list(img_dir2.glob("*.jpg"))):

Comments

[zassota]

できました、ありがとうございました！