使用python

时间:2015-08-12 11:33:47

标签: python image algorithm image-manipulation pillow

我想制定一个执行以下操作的算法,输入是左图像,输出是右图像:

重要提示:图像是动态生成的,我需要删除的水平线总是在同一位置。

enter image description here

在输出中,我并不真正关心颜色,它可能是白色的,也可能是黑色的,尽管这可能有点复杂:

enter image description here

到目前为止,我已经提出了这种修改图像的算法:

def modify_image(filename):
    img = Image.open(filename)
    img = img.convert('RGBA')

    pixdata = img.load()

    for y in xrange(img.size[1]):
        for x in xrange(img.size[0]):
            if pixdata[x, y][0] in range(155, 190): #removes horizontal lines but doesn't change them to the vertical lines color
                pixdata[x, y] = (255, 255, 255, 255)
            if pixdata[x, y][0] < 135:
                pixdata[x, y] = (0, 0, 0, 255)

    for y in xrange(img.size[1]):
        for x in xrange(img.size[0]):
            if pixdata[x, y][1] < 195:
                pixdata[x, y] = (0, 0, 0, 255)

    for y in xrange(img.size[1]):
        for x in xrange(img.size[0]):
            if pixdata[x, y][2] > 0:
                pixdata[x, y] = (255, 255, 255, 255)

    img = img.resize((1000, 300), Image.NEAREST)
    im2 = Image.new('P', img.size, 255)
    im = img.convert('P')

    img.save(base_path + "current.jpg")

modify_image (base_path + "image.gif")

这会使图像变为黑白,有时会成功将中间条与背景隔离,事实上,它不会移除水平线,也不会将颜色更改为垂直线。

请记住,颜色只是说明性和动态,所以想想任何可能的颜色。

任何建议/方法都非常重要:)

更新:设法删除水平线但不用垂直线颜色替换它们,这样只能部分解决问题(或根本不解决)

1 个答案:

答案 0 :(得分:1)

我不知道这需要多么准确,但是,由于水平条的位置是固定的,您可以将像素值从紧邻每个水平条之前的行复制到水平条的每一行。

由于渐变背景,这看起来并不完美,但它可能足以满足您的需要。查看第一个示例图像,试试这个:

import os.path
from PIL import Image

HORIZONTAL_BAND_1_Y = range(37, 64)
HORIZONTAL_BAND_2_Y = range(125, 149)

img = Image.open(filename)
img = img.convert('RGBA')
pixdata = img.load()

for band in HORIZONTAL_BAND_1_Y, HORIZONTAL_BAND_2_Y:
    for y in band:
        for x in range(img.size[0]):
            pixdata[x,y] = pixdata[x, band[0]-1]

new_filename = os.path.join(os.path.dirname(filename),
                   'new_{}'.format(os.path.basename(filename)))

img.save(new_filename)

样本输出

enter image description here