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

Question

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

此文件夹中有如图所示的 .npy 文件the .npy files

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

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

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

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

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

Answer 1

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

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)

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

Merge multiple .npy files into single .npy file

numpy

numpy-ndarray