小编Joh*_*idt的帖子

由于 keras 的 ImageDataGenerator 不适合 3D 体积，我开始为 keras 编写自己的生成器（语义分割，而不是分类！）。

1) 如果有人已经修改了 ImageDataGenerator 代码以处理 3D 体积，请分享它！这家伙已经为视频做到了。

2）根据本教程，我编写了一个自定义生成器。

import glob
import os

import keras
import numpy as np
import skimage
from imgaug import augmenters as iaa


class DataGenerator(keras.utils.Sequence):
    """Generates data for Keras"""
    """This structure guarantees that the network will only train once on each sample per epoch"""

    def __init__(self, list_IDs, im_path, label_path, batch_size=4, dim=(128, 128, 128),
                 n_classes=4, shuffle=True, augment=False):
        'Initialization'
        self.dim = dim
        self.batch_size = batch_size
        self.list_IDs = …