Pythonの既存のコードで複数の画像をまとめて処理したい

Question

現在、以下のサイトのコードを用いて画像処理を行っています。
https://qiita.com/fallaf/items/1c5387a79027b2ec64b0

1枚づつの処理は、問題なく出来たのですが、まとめてフォルダ内の画像を処理したいため、For文を使用して、コードを追加しています。

import cv2
import numpy as np
import glob
import os
from PIL import Image


files =glob.glob("/Users/guests/*")

for fname in files:
    img = cv2.imread("fname")
    ksize = 51
    blur = cv2.blur(img, (ksize, ksize))
    rij = img/blur
    index_1 = np.where(rij >= 0.98)
    index_0 = np.where(rij < 0.98)
    rij[index_0] = 0
    rij[index_1] = 1
    cv2.imwrite("rij_image.png", rij*255) # rijの値は0～1になるはずなので255倍

フォルダを指定して開いて、処理を行い、同じファイル名でフォルダに保存したいのですが、どのように実装すれば良いでしょうか？
教えて頂けますと幸いです。

Answer 1

元と同じ名前にすることも可能ですが、上書きして元の画像が消えるのでふつうはやらない方が良いでしょう。

for fname in files:
    img = cv2.imread("fname")
    ksize = 51
    blur = cv2.blur(img, (ksize, ksize))
    rij = img/blur
    index_1 = np.where(rij >= 0.98)
    index_0 = np.where(rij < 0.98)
    rij[index_0] = 0
    rij[index_1] = 1
    cv2.imwrite(fname+"rij_image.png", rij*255) # rijの値は0～1になるはずなので255倍

とすれば、別々の名前で保存できます。
あるいは、別のフォルダを作ってそこに元と同じ名前で書きこむというのもよくやります。

Comments

[tokunaga2kasa]

ありがとうございます。実行したのですが、今度はOpenCVのエラーが出てしまいました。

blur = cv2.blur(img, (ksize, ksize))
cv2.error: OpenCV(4.5.5) /Users/runner/work/opencv-python/opencv-python/opencv/modules/imgproc/src/box_filter.dispatch.cpp:446: error: (-215:Assertion failed) !_src.empty() in function 'boxFilter'

[yominet]

img = cv2.imread("fname")
「fname」を””でくくっちゃいかんのでは

[tachikoma]

f-stringを使いたいように見えました。cv2.imread(f"{fname}")としてみるのはどうでしょうか。

[tokunaga2kasa]

ありがとうございます。
>「fname」を””でくくっちゃいかんのでは
まさに、そこが原因でした！ありがとうございます！