您的位置: 首页 > 文章 > numpy 三维矩阵下采样小技巧 numpy 三维矩阵下采样小技巧 分类: 文章 • 2024-09-08 08:22:28 问题描述 我们有一个(160, 192, 224)的三维矩阵,由于其过大,跑神经网络时显存不够。此时我们的一个思路就是对其进行一个简单的下采样,即没3x3x3的小方格里只取其中间的那个数。现在的问题是如何快速高效地实现这个计算,而不是使用多个for循环。 解决方案 down_sample = rawdata[::3,::3,::3] # 这里的rawdata是(160, 192, 224)的矩阵。 一行代码搞定 结果展示 结果显示这行代码是管用的。