指定したフォルダ内の複数枚数の画像処理について

Question

以下のプログラムで入力画像から特定の色を抽出したり2値化しました。

import cv2
image = cv2.imread("picture1.jpg")
image_mask = cv2.inRange(image, (0,0,200),(10,10,255))
cv2.imwrite("mask.jpg", image_mask)
out = cv2.bitwise_and(image, image, mask = image_mask)
cv2.imwrite("red.jpg", out)

これでは一枚の画像しか処理できないのですが、指定したフォルダ内の複数枚の画像処理をして処理した結果を別の指定したフォルダに保存させることはできますでしょうか。

2行目に入力画像を指定して、4行目に2値化した画像、6行目に特定の色を抽出した結果を出力しているのでここを変更すればいいと思ったため、以下のサイトを参考にパスを記入しました
Python, OpenCVで画像ファイルの読み込み、保存（imread, imwrite）

import cv2
image = cv2.imread("Desktop\clusterling\image\picture*.jpg")
image_mask = cv2.inRange(image, (108,64,155),(255,125,255))
cv2.imwrite("Desktop\clusterling\mask\mask%s.jpg", image_mask)
out = cv2.bitwise_and(image, image, mask = image_mask)
cv2.imwrite("clusterling\gazousyoriconv\red%s.jpg", out)

しかし以下のようなエラーが発生しました。

SyntaxError: (unicode error) 'unicodeescape' codec can't decode bytes in position 2-3: truncated \UXXXXXXXX escape

変更する場所やフォルダ指定なのでパスを指定するまではよいと思うのですがどこがいけないのかよくわかりません。

よろしくお願いいたします。

Comments

[1T2R3M4]

調べたこと、試したことを質問に追記していただけませんか。
https://teratail.com/help/question-tips#questionTips1-2

[naots]

ご指摘ありがとうございます。自身で調べて試したことを追加しました。

Answer 1

例えば imread() の公式のドキュメントには

Loads an image from a file.

と書いてあり、1つのファイルから1つの画像を読み込むことがわかります。ワイルドカードに対応しているとかフォルダを指定できるとかはまったく書いてありません。複数のファイルを処理したい場合には複数回 imread() を呼ぶ必要があります。

glob() でファイルのリストを得てループするとよいでしょう。

for fname in glob.glob("Desktop\clusterling\image\picture*.jpg")
    image = cv2.imread(fname)

    image_mask = cv2.inRange(image, (108,64,155),(255,125,255))
    # 文字列操作でfnameからマスク用のファイル名を作って、imwrite()に指定する
    cv2.imwrite(..., image_mask)

    out = cv2.bitwise_and(image, image, mask = image_mask)
    # 文字列操作でfnameからred用のファイル名を作って、imwrite()に指定する
    cv2.imwrite(..., out)

Comments

[naots]

回答ありがとうございます。
以下のように修正して実行してみたらエラーは出なくなりましたが、画像処理して指定のフォルダに保存する処理が行われませんでした。

import cv2
import glob
for fname in glob.glob("Desktop/clusterling/image/picture*.jpg"):
image = cv2.imread(fname)

image_mask = cv2.inRange(image, (108,64,155),(255,125,255))
cv2.imwrite("Desktop/clusterling/mask/mask%s.jpg" % fname, image_mask)

out = cv2.bitwise_and(image, image, mask = image_mask)
cv2.imwrite("Desktop/clusterling/gazousyoriconv/red%s.jpg" % fname, out)

改善すべき点があれば教えていただきたく思います。
よろしくお願いいたします。

[int32_t]

ぱっと見ではとくに変なところは見つかりませんね。
デバッガやprintなどでどこまで期待通り動いているか検証してください。

Answer 2

ファイル名に％の文字は使えません。
そこでなにをしようとしてるんでしょう

Comments

[naots]

以下のサイトを参考に画像処理したものを%sを使うことでmask1.jpg、mask2.jpg...mask10.jpgというように番号を付けて保存できると思ったのですがここで使うのは不適切でしょうか。
https://projects.raspberrypi.org/ja-JP/projects/getting-started-with-picamera/5
よろしくお願いいたします。

[y_waiwai]

あらかじめそのようにファイル名の文字列を作ってから、そいつに渡しましょう