将多个 .npy 文件合并为单个 .npy 文件

问题描述

我有一个文件夹，其中有 100 多个 .npy 文件。此文件夹的路径是“/content/drive/MyDrive/lung_cancer/subset0/trainImages”。

该文件夹包含 .npy 文件，如图 the .npy files

每个 .npy 文件的形状是 (3,512,512)

我想将所有这些文件合并到一个名为 trainImages.npy 的文件中，以便我可以用它来训练我的 unet 模型。

我的 unet 模型接受形状 (1,512) 的输入。我将上面的 trainImages.npy 文件加载到 imgs_train 中，如下所示，将其作为输入传递给 unet 模型

imgs_train = np.load(working_path+"trainImages.npy").astype(np.float32)

有人能告诉我如何将所有这些 .npy 文件连接成一个 .npy 文件吗？谢谢。

解决方法

所以我自己找到了答案，如果有人需要，我会附上下面的代码。根据您的需要更改它..

import os
import numpy as np
path = '/content/drive/MyDrive/lung_cancer/subset0/trainImages/'
trainImages = []
for i in os.listdir(path):
  data = np.load(path+i)
  trainImages.append(data)

numpy numpy-ndarray