用Python進行圖像處理

圖像處理的研究領域,你們都知道Matlab,而今天我要給你們介紹如何使用python的Imaging庫來進行圖像處理。
    下載python的Image library(PIL):PIL-1.1.6.win32-py2.5.exe[我用的是python2.5的windows版],500多k的一個可執行文件。安裝到python目錄。
    下面咱們就能夠開始了。
    一、圖片的打開和顯示:
    python

>> import Image   #加載image庫
>> im = Image.open("c:\test.jpg")  #打開文件[支持大部分圖像文件格式]
>> im.show()                       #顯示圖像
>> im.save("hello.gif","GIF")      #保存圖像爲gif格式

    二、進行基本的處理:windows

>> im = addNoise(im,"uniform",[0,50])  #添加uniform噪聲
>> im = spacFilter("median",im)        #中值濾波                
>> im = im.filter(ImageFilter.EDGE_ENHANCE) #基本濾波之 邊緣加強
>> im = im.convert("L")              #轉換爲灰度圖像

    原圖:dom

邊緣加強後:函數


    轉換爲灰度圖像後:


    三、能夠看到,python的強大功能在於以最少的代碼實現最多的功能。
    其實PIL徹底能夠本身定義處理函數的,上面用到的spacFilter和addNoise是我從新編寫的基本圖像處理函數:spa

defspacFilter(mode,im):
    draw = ImageDraw.Draw(im)
    for i in range(0,list(im.size)[0]):
        for j in range(0,list(im.size)[1]):
            if ((i - 1) >= 0) and ((j - 1) >= 0) and ((i + 1) < list(im.size)[0]) and ((j + 1) < list(im.size)[1]):
                pixels = [im.getpixel((i + 1,j + 1)),im.getpixel((i + 1,j)),im.getpixel((i + 1,j - 1)), \
                      im.getpixel((i,j + 1)),im.getpixel((i,j - 1)), \
                      im.getpixel((i - 1,j + 1)),im.getpixel((i - 1,j)),im.getpixel((i - 1,j - 1))]
            else:
                continue
            if mode == "mean":       #3x3平均值濾波
                #print pixels
                #color = 128
                datax = list(pixels)
                color = (datax[0] + datax[1] + datax[2] + datax[3] + datax[4] + datax[5] + \
                         datax[6] + datax[7]) / 8       
            elif mode == "median":  #3x3中值濾波
                color = (pixels[0] + pixels[1] + pixels[2] + pixels[3] + pixels[4] + pixels[5] + \
                         pixels[6] + pixels[7]) / 8
            elif mode == "max":     #3x3最大值濾波
                color = max(pixels)
            elif mode == "min":     #3x3最小值濾波
                color = min(pixels)
            point = [i,j]
            draw.point(point,color)
    del draw
    return im

    和上面用到的添加噪聲函數:orm

 def addNoise(im,mode,value):    draw = ImageDraw.Draw(im)    for i in range(0,list(im.size)[0]):        for j in range(0,list(im.size)[1]):            if mode == "uniform":       #UNIFORM噪聲                rnd = random.uniform(value[0],value[1])            elif mode == "normal":          #UNIFORM噪聲                rnd = random.gauss(value[0],value[1])            elif mode == "lognormal":                rnd = random.lognormvariate(value[0],value[1])            elif mode == "negexp":                rnd = random.expovariate(value[0],value[1])            elif mode == "gamma":                rnd = random.gammavariate(value[0],value[1])            elif mode == "beta":                rnd = random.betavariate(value[0],value[1])            elif mode == "pareto":                rnd = random.paretovariate(value[0])            elif mode == "weibull":                rnd = random.weibullvariate(value[0],value[1])                           if im.mode == "RGB":                color = list(im.getpixel((i,j)))                 color[0] = color[0] + rnd                color[1] = color[1] + rnd                color[2] = color[2] + rnd                point = [i,j]                draw.point(point,tuple(color))            elif im.mode == "L":                color = im.getpixel((i,j))                color = color + rnd                point = [i,j]                draw.point(point,color)            else:                print "File type not supported!"                sys.exit(1)    del draw    return im
相關文章
相關標籤/搜索