如何加速Python PIL图像过滤功能

时间:2017-07-04 09:35:52

标签: python image-processing python-imaging-library pillow

我有两个灰度级的png。这些图像具有相同的宽度和高度。

例如:

image1 image2

我需要按以下方式过滤这些图像:当image1中的像素值不同于255且同一位置的像素值不同于255时,我想将两个像素存储在两个单独的图像中(imageFiltered1和imageFiltered2)。然后,两个过滤后的图像都会创建一个新图像,这要归功于ImageChops。

这是我拉到一起的算法:

#!/usr/bin/env python
# -*- coding: utf-8 -*-

from PIL import Image, ImageChops

def makeCustomMultiply(image1, image2):
    assert image1.size == image2.size

    imageFiltered1 = Image.new(size=image1.size, mode='L', color=255)
    imageFiltered2 = Image.new(size=image1.size, mode='L', color=255)

    for eachY in xrange(0, imageFiltered1.size[1]):
        for eachX in xrange(0, imageFiltered1.size[0]):
            pixel1 = image1.getpixel((eachX, eachY))
            pixel2 = image2.getpixel((eachX, eachY))

            if pixel1 == 255 or pixel2 == 255:
                imageFiltered1.putpixel((eachX, eachY), 255)
                imageFiltered2.putpixel((eachX, eachY), 255)
            else:
                imageFiltered1.putpixel((eachX, eachY), pixel1)
                imageFiltered2.putpixel((eachX, eachY), pixel2)

    combo = ImageChops.multiply(imageFiltered1, imageFiltered2)
    return combo

if __name__ == '__main__':

    image1 = Image.open('image1.png')
    image2 = Image.open('image2.png')

    myCustomMultiply = makeCustomMultiply(image1, image2)
    myCustomMultiply.save('myCustomMultiply.png')

它基本上是一个乘法函数,其中未显示黑色/灰色与白色。然后只有灰色到灰色相乘。

我的代码可以以某种方式改进吗? 我想避免嵌套的for循环,这会让我的代码变慢很多。每次运行程序时,此函数必须使用数百次。

由于

输出:

enter image description here

1 个答案:

答案 0 :(得分:0)

来自docs

  

<强> getpixel
  ...
  请注意,此方法相当慢;如果你需要从Python处理图像的较大部分,你可以使用像素访问对象(参见load)或getdata方法。

或者,为了加速代码的纯Python部分,请尝试使用PyPy