Question

我正在尝试在python中创建幻灯片搜索算法。我试图通过遍历坐标来索引图像。请检查下面的代码。在这里，我正在使用2 FOR LOOPS（这需要很多时间）。我想实现它，它将很快地计算出来。有什么更好的矢量化方法可以更快地实现？

width,height,depth = img.shape
wind_size = 64
xx = np.arange(0,width,wind_size)
yy = np.arange(0,height,wind_size)
for i in xx[:-1]:
    for j in yy[:-1]:
        img_w = img[i:i+wind_size,j:j+wind_size,:]
        img_w = cv2.cvtColor(img_w,cv2.COLOR_BGR2RGB)
        img_lst1.append(img_w/255.0)

Answer 1

使用here中的window_nd，然后做

window_nd(image, window = 64, steps = 64, axis = (0, 1))

这使用np.stride_tricks.as_strided创建窗口作为原始图像的视图。

或者，您可以使用skimage.util.shape.view_as_blocks

skimage.util.shape.view_as_blocks(image, (64, 64, 3))

滑动搜索窗口而不在python中使用'for'循环

1 个答案: