在 Python 中将数据加载到训练集中

Question

我需要为我的 2 个目录中的所有文件创建一个训练和测试集。但我只能选择一个文件。如何浏览所有 .npz 文件？图像存储在图像文件夹中，格式如下：[patient_id].npz. 每个 .npz 文件都包含扫描和掩码（3-d 数组），可通过以下方式访问：import numpy as nparchive = np.load('Train set/Images/patient_002.npz')train_scan = archive['scan']train_mask = archive['mask']test_set = np.load('Test set/Images/patient_000.npz')test_scan = test_set['scan']test_mask = test_set['mask']使用 test_scan.shape 和 train_mask.shape 作为 3d 矩阵：出去：(92, 92, 92)

Answer 1

您可以使用os.walk。

import os

all_files = []

for _, _, file_name in os.walk('Train set/Images/'):

    all_files.extend(file_name)

all_files = [file_name for file_name in all_files if file_name.endswith('.npz')]

然后您可以在 for 循环中读取 all_files。希望这可以帮助。