Python+pillow计算椭圆图形几何中心

2017-04-30 董付国 Python小屋 Python小屋


本文所用测试图像文件位于当前文件夹的testimages子文件夹中,并且图像以白色为背景。


from PIL import Image

import os


def searchLeft(width, height, im):

    #从左向右扫描

    for w in range(width):

        #从下向上扫描

        for h in range(height):

            #获取图像指定位置的像素颜色

            color = im.getpixel((w, h))

            if color != (255, 255, 255):

                #遇到并返回椭圆边界最左端的x坐标

                return w



def searchRight(width, height, im):

    #从右向左扫描

    for w in range(width-1, -1, -1):

        for h in range(height):

            color = im.getpixel((w, h))

            if color != (255, 255, 255):

                #遇到并返回椭圆边界最右端的x坐标

                return w

            

def searchTop(width, height, im):

    for h in range(height-1, -1, -1):

        for w in range(width):

            color = im.getpixel((w,h))

            if color != (255, 255, 255):

                #遇到并返回椭圆边界最上端的y坐标

                return h


def searchBottom(width, height, im):

    for h in range(height):

        for w in range(width):

            color = im.getpixel((w,h))

            if color != (255, 255, 255):

                #遇到并返回椭圆边界最下端的y坐标

                return h


#遍历指定文件夹中所有bmp图像文件,假设图像为白色背景,椭圆为其他任意颜色

images = [f for f in os.listdir('testimages') if f.endswith('.png')]

for f in images:

    f = 'testimages\\'+f

    im = Image.open(f)

    

    #获取图像大小

    width, height = im.size

    x0 = searchLeft(width, height, im)

    x1 = searchRight(width, height, im)

    y0 = searchBottom(width, height, im)

    y1 = searchTop(width, height, im)

    center = ((x0+x1)//2, (y0+y1)//2)


    #把椭圆中心像素画成红色

    im.putpixel(center, (255,0,0))

    #保存为新图像文件

    im.save(f[0:-4]+'_center.png')

    im.close()


测试结果:

原始图像1

运行结果:

原始图像2:

运行结果: