分割した画像を保存する際，毎

Question

機械学習用のデータの水増しのために，画像を分割して保存したいのですが，ファイル名が処理するたびに上書き保存されてしまいます．
リネームの方法も検討したのですが，どうもうまくいかず..．
ご教授いただけると幸いです．

実現したいこと

ファイル名が同じでも上書き保存されないようにしたい．
あるいはファイル名が既存のものと重複する場合はファイル名を取得できるようにしたい

該当のソースコード

Python（3.10.5）
ソースコード
from pathlib import Path
import cv2
import numpy as np

img = cv2.imread(r"ファイル保存場所")
rows = 3
cols = 3

chunks = []
for row_img in np.array_split(img,rows,axis=0):
for chunk in np.array_split(row_img,cols,axis=1):
chunks.append(chunk)
print(len(chunks))

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

#保存する
for i,chunk in enumerate(chunks,1):
save_path = output_dir.name + "/" + f"photo{i}.jpg"
cv2.imwrite(str(save_path),chunk)

Comments

[TakaiY]

> ファイル名が処理するたびに上書き保存されてしまいます
ソースコード(インデントがないので推測ですが)を見ても特に問題はなさそうですが、「処理するたび」というのは、もしかして、「ファイル保存場所」のところを別ファイル名に書き換えて複数回実施しているということでしょうか？

[Skinny-boy]

追記・修正の依頼ありがとうございます。

毎処理使うファイルを変えているので、そう言う事になると思います。

[TakaiY]

なるほど。ということは、その時は必ず前のファイルと同じディレクトリに同じファイル名で保存されるので、上書きになりますね。
で、その時に出力される分割したファイルはどうしたいのですか？
別の場所に保存する？

[Skinny-boy]

とりあえずは同じ場所に保存したいと考えています。

[TakaiY]

同じ場所に保存ということは、ファイル名を変えたいということだと思いますが、どのような名前に変えますか？

[usekay]

もとのファイル名+連番とか？

[Skinny-boy]

例えば、分割してできた既存ファイル名がphoto_50まであるとしたら、次の処理ではphoto_51からになるようにしたいです。
わかりづらくてすみません。

[TakaiY]

それ、難しいし、あまり意味があるとは思えない。
そうしたり理由は何ですか？

[Skinny-boy]

より簡単にできる方法であれば別の場所に保存する形でもいいと考えています。

いかんせん始めたばかりで何が簡単そうかの判断もつかないので、調べてやりやすそうな上記の形にした次第です…。

[TakaiY]

こういうとき、よくやるのは、元のファイル名を使ったものです。
1つのディレクトリにまとめたければ、usekayさんの書き込みにあるように、もとのファイル名+連番としたり、元のファイル名と同じ名前のフォルダに入れるなどです。
そうすることで、分割後のファイルが、あとからでもどのファイルｊを分割したものなのかすぐにわかるので、使いやすいでしょう。

Answer 1

質問でいろいろつついてしまったので、例を作ってみました。
3パターンあります。 参考になれば。

import pathlib
import re
import cv2
import numpy as np

while True:
    image_name = input(r'元のファイル名を入力してください。 :')
    image_file_path = pathlib.Path(image_name)
    if image_file_path.exists():
        img = cv2.imread(image_name)
        break

rows = 3
cols = 3

chunks = []
for row_img in np.array_split(img,rows,axis=0):
    for chunk in np.array_split(row_img,cols,axis=1):
        chunks.append(chunk)

#------------------------------
# その1: フォルダにある一番大きな番号の続きの番号を振る

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

output_dir_file_list = output_dir.glob("photo*.jpg")
num_list = []
for n in output_dir_file_list:
    m = re.match(r'photo0*(\d+).jpg', n.name)
    num_list.append(int(m.groups()[0]))
if len(num_list) == 0:
    max_num = 0
else:
    max_num = max(num_list)

#保存する
for i,chunk in enumerate(chunks,1):
    save_path = output_dir.name + "/" + f"photo{max_num + i}.jpg"
    cv2.imwrite(str(save_path),chunk)


#------------------------------
# その2: 元画像ファイル名と同じ名前のフォルダに入れる
#出力先のディレクトリを作成する
output_dir = pathlib.Path(image_file_path.stem)
output_dir.mkdir(exist_ok=True)

#保存する
for i,chunk in enumerate(chunks,1):
    save_path = output_dir.name + "/" + f"photo{i}.jpg"
    cv2.imwrite(str(save_path),chunk)

#------------------------------
# その3: 元画像ファイル名に連番を付けた名前にする
#出力先のディレクトリを作成する
output_dir = pathlib.Path("output")
output_dir.mkdir(exist_ok=True)

#保存するUファイル名は
for i,chunk in enumerate(chunks,1):
    save_path = output_dir.name + "/" + f"{image_file_path.stem}_{i}.jpg"
    cv2.imwrite(str(save_path),chunk)

Answer 2

save_path = output_dir.name + "/" + f"photo{i}.jpg"
if os.path.isfile(save_path):
    print(f"『{save_path}』はもうあるよ")
else:
    
    cv2.imwrite(str(save_path),chunk)

てこと？