pythonで画像を二値化し面積と周囲長を学習させるプログラムを作っています。

Question

###前提・実現したいこと
pythonで画像を二値化し面積と周囲長を学習させるプログラムを作っています。
1つの画像を二値化し面積を求めることができたのですが数百枚ある画像を二値化し、面積、周囲長を求め、学習させるプログラムはどうつくればいいでしょうか。

###該当のソースコード python3.6

【二値化】
def main():
    # 閾値
    t = 127

    # 入力画像の読み込み
    img = cv2.imread("img.png")

    # グレースケール変換
    gray = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)
    
    # 方法2 （OpenCVで実装）      
    ret, th2 = cv2.threshold(gray, t, 255, cv2.THRESH_BINARY)    

    # 結果を出力
    cv2.imwrite("th2.jpg", th2)


if __name__ == "__main__":
    main()  


【面積】
img = np.array(Image.open('TC001.png').convert("1"))

###補足情報(言語/FW/ツール等のバージョンなど)
python3.6
画像はパソコン内のC:\Users\LSL\定義した文字領域に入っていて、TC001.png、TC002.png、TC003.png・・・のように名前をつけてました。

Comments

[mkgrei]

学習させるというのはどういうことでしょうか。

[next_aaa]

黒い部分の面積や周囲長を学習させ文字領域を特定するプログラムを作りたいと考えております。

Answer 1

「学習」の内容がわかりませんけど、たくさんあるファイルに対応する形に発展させたいという点に関しては、ざっくり以下のような感じでしょうか。
0. def mainの中でファイル名を決め打ちしているので、これを関数呼び出し時のパラメータから読み取るように変更する。関数名もmainからもうすこし説明的なものに変える（imageanalyzeとか何とか）。
0. 注目ディレクトリの中のファイルを列挙しそのファイル名それぞれを引数として上記関数を呼び出すようなforループを作成する。

Comments

[next_aaa]

KojiDoiさん、ありがとうございます。
その通りです。1枚の画像を二値化することはできましたが、for文などを使って数百枚ある画像を二値化し、黒い部分の面積、周囲長を求め、そこから文字領域を特定させるようなプログラムを作りたいです。

[KojiDoi]

そこまでわかっているのなら、まずはマニュアルを見てforループを実際に書いてみてください。書いてみてどこか引っかかったなら、引っかかったところを具体的に説明してください。

[next_aaa]

for i in range(1, 10):
print(f'file name/img{i}.png')
1から9まで表示できますがここから二値化や面積、周囲長の部分も応用したいと思いました。

Answer 2

以下のようなプログラムのことでしょうか。

def 文字領域を得る関数(画像):
    何かしらの形で与える
    return 領域

def 面積・周囲長を得る関数(画像):
    二値化
    面積求める
    周囲長求める
    return 面積, 周囲長

X = []
Y = []
for 画像 in フォルダ:
    領域 = 文字領域を得る関数(画像)
    面積, 周囲長 = 面積・周囲長を得る関数(画像)
    X.append([面積, 周囲長])
    Y.append(領域)

モデル = 判定用のモデル
モデル.fit(X, Y)

S, L = 面積・周囲長を得る関数(新しい画像)
新しい画像の文字領域 = モデル.predict([S, L])

Comments

[next_aaa]

mkgreiさん、ありがとうございます。
1枚の画像を二値化することはできましたが、for文などを使って数百枚ある画像を二値化し、黒い部分の面積、周囲長を求め、そこから文字領域を特定させるようなプログラムを作りたいです。forで繰り返しやすいように名前をTC001.png、TC002.png、TC003.png・・・にしたんですが、プログラムをどう書いていいのかわかりませんでした。

[mkgrei]

globかos.listdirがお探しの関数かと思います。
http://www.yukun.info/blog/2008/08/python-directory-listdir-glob.html

[next_aaa]

紹介して頂いたサイトのプログラムを実行したらC:\Users\LSL\定義した文字領域内のデータをすべて表示させることができました。これをうまく二値化や面積、周囲長を求めるプログラムに応用したいと思います。