2

我试图使用 QImage 来加载图像并使用在整个图像上移动的模板子图像检查相等性。代码如下:

for i in range(image.height() - backgroundMask.height() + 1):
        for j in range(image.width() - backgroundMask.width() + 1):
            subsection = image.copy(j, i, j + backgroundMask.width() - 1, i + backgroundMask.height() - 1)
            if subsection == backgroundMask:
                print 'equality action here'
            else:
                print 'non-equality action here'

问题是它花费了太多时间来做到这一点。使用 Python Imaging Library 的类似操作太快了。两个主要操作是 copy() 和 operator==() 。我认为大部分时间都花在了 copy() 上,因为它只在那里执行复制。如果它只是一个懒惰的写时复制操作,那么它会更快。

有什么方法可以更快地做到这一点?

4

1 回答 1

0

更快的方法是手动比较像素 - 你正在做的副本是浪费的。假设您要查找 backgroundMask 作为“图像”的子图像。你从左上角开始。现在您发现图像的像素 (0, 0) 与 backgroundMask 的 (0, 0) 不匹配。如果您手动比较像素,您只需继续到图像的 (0, 1) 并将其与 (0, 0) 进行比较,依此类推。但是在您的情况下,您已经浪费了很多时间来复制宽度 x 高度像素。

start = time.time()
for i in xrange(image.height() - backgroundMask.height() + 1):
    for j in xrange(image.width() - backgroundMask.width() + 1):
        success = True
        for y in xrange(backgroundMask.height()):
            for x in xrange(backgroundMask.width()):
                if image.pixel(j + x, i + y) != backgroundMask.pixel(x, y):
                    success = False
                    break
            if not success:
                break

        if success:
            print 'match'
        else:
            print 'no match'

诚然,Python 中的每像素访问速度很慢,并且相等运算符是用 C 编写的。但它仍然比您发布的要快得多。对于我尝试过的图像,你的代码用了 27 秒,而我的用了 0.8 秒。

但是,最好的解决方案可能是将 QImage 转换为 PIL 图像,如果该功能在那里实现的话。QImages 和 PIL 图像之间的转换很简单并且有据可查。

于 2011-06-05T10:33:16.847 回答