[Python][画像処理]For分で編集した複数画像を出力したい

Question

下記のコードで、複数画像の編集を行うことができました。
しかし、実装後の出力先(outputフォルダ)に一枚しか出力画像がありません。

▼ディレクトリ構造
user
・|-ocr
・・|-test_data
・・|・|-画像1.jpeg
・・|・|-画像2.jpeg
・・|・・・
・・|
・・|-output

import cv2 
import numpy as np
import glob

images = glob.glob('user/../ocr/test_data/*')
for im_n in images:

    img = cv2.imread(im_n)
    mark = np.zeros_like(img)

    mark = cv2.putText(cv2.UMat(mark), 'ABCDEF', (50, 300), cv2.FONT_HERSHEY_SIMPLEX, 2.0, (255, 255, 255), 5, cv2.LINE_AA)
    new_img = cv2.addWeighted(img, 1, mark, 0.3, 0)
    cv2.imwrite('user/../ocr/output/im_n.jpeg',new_img)

###試したこと
globモジュールを使用して実装するのが間違いと考え、osライブラリを使用して現在試みてます。

path = "user/../ocr/test_data"
file = os.listdir(path)

files_file = [im_n for im_n in file if os.path.isfile(os.path.join(path, im_n))]

print(files_file)

上記ですべてのファイル名を出力することはできましたが、画像ファイルとして入力し、編集して、編集されたすべての画像ファイルをOutputフォルダに出力するということができません。

globモジュールとOsライブラリをどちらを使用する形でも問題ございませんので
どなたか、知恵を貸して頂きたく存じます。宜しくお願い致します。(ちなみに、ファイル名にこだわりはありません)

Answer 1

出力の画像の名前をループ一周ごとに替えないと、一枚の画像が上書きされ続けちゃいます。

import cv2 
import numpy as np
import glob

images = glob.glob('user/../ocr/test_data/*')
for i, im_n in enumerate(images):

    img = cv2.imread(im_n)
    mark = np.zeros_like(img)

    mark = cv2.putText(cv2.UMat(mark), 'ABCDEF', (50, 300), cv2.FONT_HERSHEY_SIMPLEX, 2.0, (255, 255, 255), 5, cv2.LINE_AA)
    new_img = cv2.addWeighted(img, 1, mark, 0.3, 0)
    cv2.imwrite('user/../ocr/output/im_n_{}.jpeg'.format(i),new_img)

でどうでしょ。

Comments

[kobayashy_kazu]

ご回答ありがとうございます、解決することが出来ました...（泣）

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

Answer 2

import cv2 
import numpy as np
import glob
import os

path = "user/../ocr/test_data"
file = os.listdir(path)

files_file = [im_n for im_n in file if os.path.isfile(os.path.join(path, im_n))]

print(files_file)

for im_n in files_file:

    img = cv2.imread(im_n)
    mark = np.zeros_like(img)

    mark = cv2.putText(cv2.UMat(mark), 'ABCDEF', (50, 300), cv2.FONT_HERSHEY_SIMPLEX, 2.0, (255, 255, 255), 5, cv2.LINE_AA)
    new_img = cv2.addWeighted(img, 1, mark, 0.3, 0)
    cv2.imwrite('user/../ocr/output/' + im_n, new_img)

とした場合どうでしょうか？

Comments

[kobayashy_kazu]

im_n.jpegという名前の画像が一枚だけ出力されました...

[退会済みユーザー]

修正しました

[kobayashy_kazu]

ご回答ありがとうございます。
-----------
import cv2
import numpy as np
import glob
import os

path = "user/../ocr/test_data"
file = os.listdir(path)

files_file = [im_n for im_n in file if os.path.isfile(os.path.join(path, im_n))]

print(files_file)

for im_n in files_file:

img = cv2.imread(im_n)
mark = np.zeros_like(img)

mark = cv2.putText(cv2.UMat(mark), 'ABCDEF', (50, 300), cv2.FONT_HERSHEY_SIMPLEX, 2.0, (255, 255, 255), 5, cv2.LINE_AA)
new_img = cv2.addWeighted(img, 1, mark, 0.3, 0)
cv2.imwrite('user/../ocr/output/' + im_n, new_img)
-------------
TypeError: UMat() missing required argument 'ranges' (pos 2)
上記のコードだとこちらの出力エラー内容が出てきました。現在自分の方でも打開策模索中でございます

[退会済みユーザー]

mark = cv2.putText(cv2.UMat(mark), 'ABCDEF', (50, 300), cv2.FONT_HERSHEY_SIMPLEX, 2.0, (255, 255, 255), 5, cv2.LINE_AA)
を
mark = cv2.putText(mark, 'ABCDEF', (50, 300), cv2.FONT_HERSHEY_SIMPLEX, 2.0, (255, 255, 255), 5, cv2.LINE_AA)
にした場合どうでしょうか。
（cv2.UMat(mark)　→　mark　にする）

[kobayashy_kazu]

qnoirさん、何度もご提案ありがとうございましたｍｍ

こちらで解決することが出来ました！
import cv2
import numpy as np
import glob

images = glob.glob('user/../ocr/test_data/*')
for i, im_n in enumerate(images):

img = cv2.imread(im_n)
mark = np.zeros_like(img)

mark = cv2.putText(cv2.UMat(mark), 'ABCDEF', (50, 300), cv2.FONT_HERSHEY_SIMPLEX, 2.0, (255, 255, 255), 5, cv2.LINE_AA)
new_img = cv2.addWeighted(img, 1, mark, 0.3, 0)
cv2.imwrite('user/../ocr/output/im_n_{}.jpeg'.format(i),new_img)