numpy 三维矩阵下采样小技巧

问题描述

我们有一个(160, 192, 224)的三维矩阵，由于其过大，跑神经网络时显存不够。此时我们的一个思路就是对其进行一个简单的下采样，即没3x3x3的小方格里只取其中间的那个数。现在的问题是如何快速高效地实现这个计算，而不是使用多个for循环。

解决方案

down_sample = rawdata[::3,::3,::3] # 这里的rawdata是(160, 192, 224)的矩阵。
一行代码搞定

结果展示

结果显示这行代码是管用的。