Question

我正在尝试将numpy数组[width x height x color]裁剪为预定义的较小尺寸。

我找到了应该做我想要的东西，但它只适用于[width x height]数组。我不知道如何使它适用于具有额外颜色维度的numpy数组。

Answer 1

使用numpy，您可以使用范围索引。假设您有一个列表x[]（单维），您可以将其索引为x[start:end]，这称为切片。

切片可以用于更高的尺寸，如

x[start1:end1][start2:end2][start3:end3]

这可能就是你要找的东西。

虽然记住这并没有生成新的数组（即它不会复制）。对此的更改将反映为x。

Answer 2

从您链接的问题来看，代码只是一个小小的变化：

def crop_center(img,cropx,cropy):
    y,x,c = img.shape
    startx = x//2 - cropx//2
    starty = y//2 - cropy//2    
    return img[starty:starty+cropy, startx:startx+cropx, :]

所有添加的内容是最后一行末尾的另一个:和解包形状的（未使用的）c。

>>> img
array([[[ 18,   1,  17],
        [  1,  13,   3],
        [  2,  17,   2],
        [  5,   9,   3],
        [  0,   6,   0]],

       [[  1,   4,  11],
        [  7,   9,  24],
        [  5,   1,   5],
        [  7,   3,   0],
        [116,   1,  55]],

       [[  1,   4,   0],
        [  1,   1,   3],
        [  2,  11,   4],
        [ 20,   3,  33],
        [  2,   7,  10]],

       [[  3,   3,   6],
        [ 47,   5,   3],
        [  4,   0,  10],
        [  2,   1,  35],
        [  6,   0,   1]],

       [[  2,   9,   0],
        [ 17,  13,   4],
        [  3,   0,   1],
        [ 16,   1,   3],
        [ 19,   4,   0]],

       [[  8,  19,   3],
        [  9,  16,   7],
        [  0,  12,   2],
        [  4,  68,  10],
        [  4,  11,   1]],

       [[  0,   1,  14],
        [  0,   0,   4],
        [ 13,   1,   4],
        [ 11,  17,   5],
        [  7,   0,   0]]])
>>> crop_center(img,3,3)
array([[[ 1,  1,  3],
        [ 2, 11,  4],
        [20,  3, 33]],

       [[47,  5,  3],
        [ 4,  0, 10],
        [ 2,  1, 35]],

       [[17, 13,  4],
        [ 3,  0,  1],
        [16,  1,  3]]])

Answer 3

numpy适用于任何维度

import numpy as np
X = np.random.normal(0.1, 1., [10,10,10])
X1 = X[2:5, 2:5, 2:5]
print(X1.shape)

最后一个打印语句产生[3,3,3]数组。

中心裁剪一个numpy阵列

3 个答案: