Question

我知道使用OpenCV和C ++迭代像素并访问它们的值。现在，我正在尝试学习python ，我试图在python中做同样的事情。但是当我运行以下代码时，显示图像需要花费很多时间（约7-10秒）。即使在显示图像之后，脚本也会继续运行几秒钟。

我发现了一个类似的问题here at SO但是我无法理解我如何在我的情况下使用numpy（因为我是python中的初学者）以及是否真的需要它？

代码说明：我只想将黑色像素放在图像的左侧和右侧。

import numpy as np
import cv2 as cv

#reading an image
img = cv.imread('image.jpg')
height, width, depth = img.shape

for i in range(0, height):
    for j in range(0, (width/4)):
        img[i,j] = [0,0,0]  

for i in range(0, height):
    for j in range(3*(width/4), width):
        img[i,j] = [0,0,0]        

cv.imshow('image',img)

cv.waitKey(0)

Answer 1

（注意：我不熟悉opencv，但这似乎是numpy问题）

＆＃34;非常慢＆＃34;部分是你在python字节码中循环，而不是让numpy以C速度循环。

尝试直接分配到（3维）切片，该切片会屏蔽您想要清零的区域。

import numpy as np

example = np.ones([500,500,500], dtype=np.uint8)

def slow():
     img = example.copy()
     height, width, depth = img.shape
     for i in range(0, height):             #looping at python speed...
         for j in range(0, (width//4)):     #...
             for k in range(0,depth):       #...
                 img[i,j,k] = 0
     return img


def fast():
     img = example.copy()
     height, width, depth = img.shape
     img[0:height, 0:width//4, 0:depth] = 0 # DO THIS INSTEAD
     return img 

np.alltrue(slow() == fast())
Out[22]: True

%timeit slow()
1 loops, best of 3: 6.13 s per loop

%timeit fast()
10 loops, best of 3: 40 ms per loop

以上显示左侧归零;为右侧做同样的事情是读者的练习。

如果numpy切片语法让你烦恼，我建议你阅读indexing docs。

使用python（OpenCV），图像中像素的迭代速度非常慢

1 个答案: