大津の二値化を流用した閾値設定

Question

前提

Pythonで二群間を最大に分離する閾値設定をしたいです。
0から256でヒストグラムを作製し、最も2群を分離する値を、クラス間分散が最大とする値を閾値として設定します。
初心者のもので、まったくなにからまちがえたかがわかりません。

実現したいこと

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

エラーメッセージ

該当のソースコード

import cv2
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt


#大津の手法
def threshold_otsu(data, min_value=0, max_value=255):


    #ヒストグラムの算出
    hist = [np.sum(data==i) for i in range(256)]

    s_max=(0,-10)

    for th in range(256):
        
        # クラス1とクラス2の画素数を計算
        n1 = sum(hist[:th])
        n2 = sum(hist[th:])
        

        # クラス1とクラス2の画素値の平均を計算
        if n1 == 0 : mu1 = 0
        else : mu1 = sum([i * hist[i] for i in range(0,th)]) / n1   
        if n2 == 0 : mu2 = 0
        else : mu2 = sum([i * hist[i] for i in range(th, 256)]) / n2


        # クラス間分散の分子を計算
        s = n1 * n2 * (mu1 - mu2) ** 2

        # クラス間分散の分子が最大のとき、クラス間分散の分子と閾値を記録
        if s > s_max[1]:
            s_max = (th, s)
    
    # クラス間分散が最大のときの閾値を取得
    t = s_max[0]
    
    #cv2.imwrite()
    print(t)
    
    return data


#ヒストグラム化するデータの読み込み

data = pd.read_csv("random2line.csv")
df=pd.read_csv("random2line.csv", header=None)
data=df.values[:,1:]

plt.hist(data, bins=256)
plt.savefig("C:\pythonlab\lab\output.jpg")
plt.show()

th = threshold_otsu(data)


#出力
0

備考

二群間を分離する値が0なわけはないのですが、0になってしまいます。

用いているデータは

1	2
0~100	155~255

でそれぞれ1000個ずつランダムに作成した数字を用いています。
この2群を最大に分離する閾値を求めたいです。

Comments

[y_waiwai]

大津の二値化、というのはいったいなんでしょうか。
詳しく説明しましょう

[rmiurmiu]

失礼しました。
画像処理の際によく使われる手法で、自動で画像二値化における閾値を算出できる手法になります。
下記参考においておきます。
https://algorithm.joho.info/image-processing/otsu-thresholding/

Answer 1

dataが整数でない場合次のようにして解決可能です．

- hist = [np.sum(data==i) for i in range(256)]
+ hist, bins = np.histogram(data, bins = 256)

ちなみに提示した値域で一様分布を作成したところ101で閾値が導出されました．

Comments

[rmiurmiu]

解決しました！
ありがとうございます。