python - NumPy/PyTorch 提取图像子集

Question

在 Numpy 中，给定一堆大小为 (N,hl,wl) 的大图像 A，坐标 x 为 size(N)，y 为 size( N)我想要获得尺寸较小的图像(N,16,16)

在 for 循环中，它看起来像这样:

B=numpy.zeros((N,16,16))
for i in range(0,N):
  B[i,:,:]=A[i,y[i]:y[i]+16,x[i]:x[i]+16]

但是我可以仅通过索引来做到这一点吗？额外问题:这个索引也能在 pytorch 中工作吗？如果不是，我该如何在那里实现它？

Answer 1

在 numpy 中，切片非常简单，相同的逻辑也适用于 pytorch 示例。例如

imgs = np.random.normal(size=(16,24,24))
imgs[:,0:12,0:12].shape
imgs_tensor = torch.from_numpy(imgs)
imgs_tensor[:,0:12,0:12].size()

其中切片中的第一个:表示选择批处理中的所有图像。第二个和第三个:表示高度和宽度的切片。