ディレクトリ内の画像（JPG)全てに対して処理を行う方法

Question

現在、ディレクトリ内にある画像に対して処理を行い、カレントディレクトリに別のdataと呼ばれるディレクトリを作り、そこに保存したいと考えています。

具体的に、サーモグラフィー画像をバウンディングボックスで囲ったものを新たな画像とし、dataフォルダに保存したいです。

私が書いたコードではどうしても、全てに処理を行い、保存することができません。
また、以下に載せるコードで実行すると永久ループになってしまいます。

私が思い描いている様な処理を教えてください。初心者の私では難しすぎます。

import cv2
import numpy as np
import cv2
import glob

def contour_extraction(image):
    hsv = cv2.cvtColor(image, cv2.COLOR_BGR2HSV)
    binary = cv2.inRange(hsv, (20, 0, 0), (50, 255, 255))
    # 輪郭抽出
    contours = cv2.findContours(binary, cv2.RETR_TREE, cv2.CHAIN_APPROX_NONE)[1]
    # 面積が一定以上の輪郭のみ残す。
    area_thresh = 10000
    contours = list(filter(lambda x: cv2.contourArea(x) > area_thresh, contours))
    create_box(contours)

def create_box(contours):
    for cnt in contours:
        # 輪郭に外接する長方形を取得する。
        x, y, width, height = cv2.boundingRect(cnt)
        print(x, y, width, height)

        # 描画する。
        cv2.rectangle(image, (x, y), (x + width, y + height), color=(0, 255, 0), thickness=2)
        for images in image:
            cv2.imwrite('Users/XXXX/Desktop/サーモ/data/hogehoge.jpg', images)
      


path = './*.jpg'
files = glob.glob(path)
for f in files:
    print(f)
    image = cv2.imread(f)
    contour_extraction(image)
print('end')

どうかよろしくお願い致しましす。

Answer 1

import cv2
import numpy as np
import cv2
import glob

img_num = 0 #画像のナンバリング用変数

def contour_extraction(image):
    hsv = cv2.cvtColor(image, cv2.COLOR_BGR2HSV)
    binary = cv2.inRange(hsv, (20, 0, 0), (50, 255, 255))
    # 輪郭抽出
    contours = cv2.findContours(binary, cv2.RETR_TREE, cv2.CHAIN_APPROX_NONE)[1]
    # 面積が一定以上の輪郭のみ残す。
    area_thresh = 10000
    contours = list(filter(lambda x: cv2.contourArea(x) > area_thresh, contours))
    create_box(contours)

def create_box(contours):
    global img_num

    for cnt in contours:
        # 輪郭に外接する長方形を取得する。
        x, y, width, height = cv2.boundingRect(cnt)
        print(x, y, width, height)

        # 描画する。
        cv2.rectangle(image, (x, y), (x + width, y + height), color=(0, 255, 0), thickness=2)
        #for images in image:　ここでのループ処理は必要ないように思えます
        save_name = 'Users/XXXX/Desktop/サーモ/data/' +  img_num + '.jpg' #画像の保存名を定義します
        cv2.imwrite(save_name, image)
        
        img_num += 1



path = './*.jpg'
files = glob.glob(path)
for f in files:
    print(f)
    image = cv2.imread(f)
    contour_extraction(image)
print('end')

時間がたった質問ですが、参考に慣れば嬉しいです。

Answer 2

dataというディレクトリは先に作っておいて、保存するときのファイル名をdata/hogehoge.jpgみたいになるようにすると意図したことができるかと思います。

Comments

[rgb_nk]

コードを書き換えました。
dataというディレクトリを作り、
cv2.imwriteでpathを
'Users/XXXX/Desktop/サーモ/data/hogehoge.jpg'
として、runさせてみたのですが、永久ループは改善されたものの、保存はされていませんでした。
どこが間違っているのか、具体的にご教授いただけないでしょうか？