本文所用测试图像文件位于当前文件夹的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:
运行结果: