在 Python 中循环遍历帧像素的更快方法?

时间:2021-07-13 01:06:22

标签: python opencv video-streaming python-imaging-library hsl

试图从一帧输入视频中找到每像素的 HSL 值,但它花费了太多时间大约 0.2 秒,但希望将时间减少到至少 0.1 秒以内。

def colorize(im, h, s, l_adjust):
    result = Image.new('RGBA', im.size)
    pixin = im.load()
    pixout = result.load()
    for y in range(im.size[1]):
        for x in range(im.size[0]):
            currentR, currentG, currentB = pixin[x, y][0]/255 , pixin[x, y][1]/255, pixin[x, y][2]/255
            #luminance
            lum = (currentR * 0.2126) + (currentG * 0.7152) + (currentB * 0.0722)
            if l_adjust > 0:
                lum = lum * (1 - l_adjust)
                lum = lum + (1.0 - (1.0 - l_adjust))
            else:
                lum = lum * (l_adjust + 1)
            l = lum
            r, g, b = colorsys.hls_to_rgb(h, l, s)
            r, g, b = int(r * 255.99), int(g * 255.99), int(b * 255.99)
            pixout[x, y] = (r, g, b, 255)
    return result

不使用这两个循环,有没有更快的方法来做到这一点?寻找类似 LUT(Look up table) 或使用 NumPy 快捷方式的东西来避免这 2 个循环。谢谢

0 个答案:

没有答案