フォルダ内のファイルに対して一括処理し、結果を別フォルダに同名称+別拡張子で保存

Question

前提・実現したいこと

特定のフォルダ内のjpgファイルを順番に読み込み、ヒストグラム結果をcsvとpngで特定のフォルダにファイル名を指定して保存することを試みています。

読み込みフォルダ
1.jpg
2.jpg　・・・

出力フォルダ（output）
1.png
1.csv
2.png
2.csv　・・・

といったように読み込みと保存をしたいです。

下記のコードの不適切な部分を教えていただけないでしょうか。

発生している問題・エラーメッセージ

エラーは発生していませんが、結果が上手くいきません。

該当のソースコード

import cv2
import numpy as np
from matplotlib import pyplot as plt
import glob

files = glob.glob('*.jpg')

for f in files: 
 img = cv2.imread(f)
 color = ('b','g','r')
 hists = []
 for i, col in enumerate(color):
     hist = cv2.calcHist([img], [i], None, [256], [0, 256])
     hists.append(hist.squeeze(axis=-1))
     plt.plot(hist, color=col)
     plt.xlim([0, 256])   
plt.savefig('output/f.png')
#plt.show()


np.savetxt("output/f.csv",hists,delimiter=",")
#print(hists)

試したこと

検索するとglob.glob()を使用することでフォルダ内の全ファイルを読み込むことが出来ることが分かり、上記のとおり記載しております。
ただ、実行するとoutputフォルダに、一番最初の画像データに対するf.csvというファイルが一つと、全画像データに対する結果が一枚の画像にまとまったf.pngが一つできて終了してしまいます。

補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

Answer 1

正規表現を使ってファイル名を取得する方法です。
例だと同じ階層のようですが、一応ディレクトリ内の画像を取得した場合も想定しています。
適当な画像で試してるのでCSVやJPGの結果が正しいのかはよくわかっていません。

import cv2
import numpy as np
from matplotlib import pyplot as plt
import glob
import re

files = glob.glob('*.jpg')

for f in files: 
    img = cv2.imread(f)
    color = ('b','g','r')
    hists = []
    for i, col in enumerate(color):
        hist = cv2.calcHist([img], [i], None, [256], [0, 256])
        hists.append(hist.squeeze(axis=-1))
        plt.plot(hist, color=col)
        plt.xlim([0, 256])
    fnm = re.search(r'[^/\.]+(?=.jpg)', f).group(0)
    plt.savefig(f'output/{fnm}.png')
    np.savetxt(f"output/{fnm}.csv",hists,delimiter=",")

追記
pltに対して追記されてるようですね。

for f in files: 
    img = cv2.imread(f)
    color = ('b','g','r')
    hists = []
    fig = plt.figure()
    for i, col in enumerate(color):
        hist = cv2.calcHist([img], [i], None, [256], [0, 256])
        hists.append(hist.squeeze(axis=-1))
        plt.plot(hist, color=col)
        plt.xlim([0, 256])
    fnm = re.search(r'[^/\.]+(?=.jpg)', f).group(0)
    plt.savefig(f'output/{fnm}.png')
    np.savetxt(f"output/{fnm}.csv",hists,delimiter=",")

これでどうでしょうか。

Comments

[Kkeguri]

ご回答いただきありがとうございます。
画像データが複数枚あり、そのヒストグラムをグラフ化したものが、どんどん重なっていってしまうようです。。。
一枚目の画像に対するヒストグラムグラフ（png）は、その画像に対するヒストグラムのグラフなのですが、
二枚目の画像に対するヒストグラムグラフ（png）が、一枚目の画像と二枚目の画像の両方のヒストグラム値がグラフ化されてしまいます。。
初心者で原因がよく分からないのですが、なぜでしょうか。。。

ヒストグラムを数値化したcsvは画像ファイルごとに保存されており、問題ありません。

[Kkeguri]

ご回答が遅くなりすみません。
追記いただいたもので問題なく出来ました。
ありがとうございました。

Answer 2

plt.savefig('output/f.png')

ここんとこ、plt.savefig('output/' + f[:-3] + 'png') ではいかが?
※ np.savetxt("output/f.csv",hists,delimiter=",") も同様

Comments

[Kkeguri]

ご回答いただきありがとうございます。
上記の方と同様ですが、
画像データが複数枚あり、そのヒストグラムをグラフ化したものが、どんどん重なっていってしまうようです。。。
一枚目の画像に対するヒストグラムグラフ（png）は、その画像に対するヒストグラムのグラフなのですが、
二枚目の画像に対するヒストグラムグラフ（png）が、一枚目の画像と二枚目の画像の両方のヒストグラム値がグラフ化されてしまいます。。
初心者で原因がよく分からないのですが、なぜでしょうか。。。

ヒストグラムを数値化したcsvは画像ファイルごとに保存されており、問題ありません。

[episteme]

一枚描くごとに plt をクリアしてないんじゃないかしら。

[Kkeguri]

ご回答が遅くなりすみません。
上記の方の回答も参考にしながら、問題無くできました。
ありがとうございました。