opencv颜色空间转换方式对比

opencv颜色空间转换方式对比

import cv2
import time


def timeit(func):
    def wrapper(a):
        t_start = time.time()
        for i in range(10000):
            func(a)
        t_end = time.time()
        print(str(func), t_end - t_start)

    return wrapper


@timeit
def convert(bgr):
    rgb = bgr[:, :, ::-1]


@timeit
def cv2_convert(bgr):
    rgb = cv2.cvtColor(bgr, cv2.COLOR_BGR2RGB)


if __name__ == "__main__":
    bgr = cv2.imread("1.jpg")
    cv2_convert(bgr)
    convert(bgr)

'''result:
 5.07412052154541
 0.0039899349212646484'''

可以看出直接转换的情况比cv2.cvtColor的方法快出许多