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

Stack Overflow用户

提问于 2021-01-29 17:02:44

回答 1查看 134关注 0票数 0

我有一个文件夹，其中包含100+ .npy文件。此文件夹的路径为'/content/drive/MyDrive/lung_cancer/subset0/trainImages'.

此文件夹包含图像the .npy files中所示的.npy文件

每个.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文件？谢谢。

numpy

numpy-ndarray

回答 1

Stack Overflow用户

回答已采纳

发布于 2021-04-30 21:27:12

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

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)

票数 0

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/65951501

复制

相似问题

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

回答 1

Stack Overflow用户

社区

活动

圈层

关于

腾讯云开发者

热门产品

热门推荐

更多推荐

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

回答 1

Stack Overflow用户

社区

活动

圈层

关于

腾讯云开发者

热门产品

热门推荐

更多推荐

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