輝度の標準偏差を用い、閾値設定し、グループ分けしたい(画像保存でうまくいってない)

Question

前提・実現したいこと

ファイル内の画像の輝度の標準偏差を出し、一定の値以下かそうでないかで二つのグループに分け、
二つのファイルに分けて保存したい。

発生している問題

画像の読み込みができていて、標準偏差は出ているが、肝心の保存ができていない。

###質問
画像の保存(グループ分け)がうまくいっていないので、どうすれば改善できそうかご教授よろしくお願いいたします。
pythonもそこまで詳しくないので、こここうしたほうがより効率的で良いんじゃないか等もあればぜひ教えてください。
説明がへたくそなので、もしわからないことがあれば、質問ください。

＜＜追記＞＞
titleが"C:\Users\potatp\Desktop\brightness\data\title_1.png" となっているのが要因見たいです。
title は "title_1" にして print_stat 関数に渡したいのですが、どうすれば改善できるでしょうか。

該当のソースコード

import numpy as np
import glob
import scipy.stats as sstats
import cv2
import os

def print_stat(fname,title):

    img_name = fname
    img = cv2.imread(fname,1)
    # Grayscale
    img2 = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)
    # 一次元配列化
    img2 = np.array(img).flatten()
    
    print("---", img_name)
    std = np.std(img2)             # 標準偏差
    print("標準偏差  : ", std)
    
    if std <= 25:
        cv2.imwrite(f"C:\Users\potato\Desktop\brightness\low\low_{title}.png",img)
    else:
        cv2.imwrite(f"C:\Users\potato\Desktop\brightness\high\high_{title}.png",img)
    
files = glob.glob('C:\Users\potato\Desktop\brightness\data\*.png')
for file in files:
    title, ext = os.path.splitext(file)
    print_stat(file,title)

Comments

[melian]

title, ext = os.path.splitext(file) ですが、例えばファイル名が C:\\Users\\nishikawa\\Desktop\\brightness\\data\\title_1.png とすると、('C:\\Users\\nishikawa\\Desktop\\brightness\\data\\title_1', '.png') となってしまいます。title は "title_1" にして print_stat 関数に渡したい、ということでしょうか？

[potapotapotato]

なるほど！
それで保存できてなかったのですね...
そうですね。title_1にしたいです。
その場合どうすればよいでしょうか？

[jbpb0]

> title は "title_1" にして

これ見てください
https://note.nkmk.me/python-os-basename-dirname-split-splitext/
の「ファイル名を取得: os.path.basename()」の「拡張子なしのファイル名」

[jbpb0]

img2 = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)
で、わざわざカラーのデータからグレースケールのデータを作ってるのに、その次の
img2 = np.array(img).flatten()
では、カラーのデータをまた使ってますけど、それは意図通りでしょうか？

[potapotapotato]

サイトを参考に
title, ext = os.path.splitext(file)
を
title= os.path.splitext(os.path.basename(file))[0]
にしたところいけました！ありがとうございます！

[potapotapotato]

>img2 = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)
で、わざわざカラーのデータからグレースケールのデータを作ってるのに、その次の
img2 = np.array(img).flatten()
では、カラーのデータをまた使ってますけど、それは意図通りでしょうか？

ミスでした！
ご指摘ありがとうございます。

Answer 1

title は "title_1" にして

これ見てください
Pythonでパス文字列からファイル名・フォルダ名・拡張子を取得、結合
の「ファイル名を取得: os.path.basename()」の「拡張子なしのファイル名」

　
また、質問内容とは直接は関係ありませんが、

    img2 = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)

で作成したグレースケールのデータから標準偏差を計算するのなら、下記を変更する必要があります

    img2 = np.array(img).flatten()

↓ 変更

    img2 = np.array(img2).flatten()

Answer 2

os.path.basename を使います。

for file in files:
    title, ext = os.path.basename(file).split('.')
    print_stat(file,title)