用Python進行圖像處理


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

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

    2、進行基本的處理:

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

    原圖:

邊緣增強后:


    轉換為灰度圖像后:


    3、可以看到,python的強大功能在於以最少的代碼實現最多的功能。
    其實PIL完全可以自己定義處理函數的,上面用到的spacFilter和addNoise是我重新編寫的基本圖像處理函數:

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

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

 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


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM