Загрузка данных в тренировочный набор в Python

Question

Мне нужно создать обучающий и тестовый набор для всех файлов в моих 2 каталогах. Но я могу выбрать только один файл. Как go просмотреть все файлы .npz?

Изображения хранятся в папке изображений в следующем формате: [patient_id].npz. Каждый файл .npz содержит как скан, так и маску (трехмерные массивы), доступные для:

import numpy as np

archive = 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-матриц:

Out:

(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')]

И затем вы можете прочитать все_файлы в a для l oop. Надеюсь, это поможет.