[Python]代碼

#coding=utf8

"""

大津法閾值分割:A Threshold Selection Method from Gray-Level Histograms

需要PIL庫

"""

import Image

import sys

def sigma(im,i,debug =False):

"""

閾值為i時,圖像im中兩組的方差

"""

c0_p_num = sum(im.histogram()[:i+1])#灰度<=k的像素個數

c1_p_num = sum(im.histogram()[i+1:])#灰度>k的像素個數

#計算兩部分的總灰度

c0_g_sum = 0

for j in range(1,i+1):

c0_g_sum += j*im.histogram()[j]

#end for j

c1_g_sum = 0

for j in range(i+1,255):

c1_g_sum += j*im.histogram()[j]

#end for j

#計算兩部分的各自平均灰度

try:

u0 = 1.0*c0_g_sum/c0_p_num

u1 = 1.0*c1_g_sum/c1_p_num

#計算兩部分的像素比例

w0 = 1.0*c0_p_num/(c0_p_num+c1_p_num)

except:

#可能有的圖像沒有太高或太低灰度的像素

return 0

w1 = 1.0 - w0

u = (u0-u1)**2

new_sigma = w0 * w1 *u

if debug:

print "%d:\tw0=%f,w1=%f,new_sigma=%f" %(i,w0,w1,new_sigma)

return new_sigma

def OtsuThreshold(im,debug = False):

"""

線性查找最大方差,可以考慮用別的搜索算法

"""

g_level = 0

g_sigma = 0

for i in range(1,255):

new_sigma = sigma(im,i,debug)

if g_sigma

g_sigma = new_sigma

g_level = i

#end for i

return g_level, g_sigma

if __name__=="__main__":

if len(sys.argv)>1:

src_file = sys.argv[1]

else:

print "USAGE:%s src_file [des_file]" %sys.argv[0]

sys.exit(1)

des_file = "OTSU_" + src_file

if len(sys.argv)>2:

des_file = sys.argv[2]

im = Image.open(src_file).convert("L")

#大津法找閾值

debug = True

threshold,max = OtsuThreshold(im,debug)

print threshold,max

#根據閾值,分割之

im = im.point(lambda p: p > threshold and 255)

im.save(des_file)

print "DONE: %s--->%s" %(src_file,des_file)

版權申明：本站文章部分自網絡，如有侵權，請聯系：west999com@outlook.com

特別注意：本站所有轉載文章言論不代表本站觀點！

本站所提供的圖片等素材，版權歸原作者所有，如需使用，請與原作者聯系。