【发布时间】:2021-06-07 14:48:42
【问题描述】:
我正在尝试在 keras 中创建自定义数据生成器 -
class DataGenerator_cnn(keras.utils.Sequence):
'Generates data for Keras'
def __init__(self, list_IDs, labels, batch_size=5, dim=(30,128,157),
n_classes=1, shuffle=True):
'Initialization'
self.dim = dim
self.batch_size = batch_size
self.labels = labels
self.list_IDs = list_IDs
self.n_classes = n_classes
self.shuffle = shuffle
self.on_epoch_end()
但是,它在执行时显示错误
ImportError: cannot import name 'Sequence' from 'keras.utils' (/usr/local/lib/python3.7/dist-packages/keras/utils/init.py)
我在 google colab 上使用最新版本的 keras (2.5.0)。请帮忙
【问题讨论】:
-
您应该分享更多代码来帮助您。无论如何,您可能会尝试从错误的路径导入
Sequential,例如:from keras.utils import Sequential。但是您应该尝试像这样导入:from keras import Sequential -
这对我有用:
from tensorflow.keras.utils import Sequence
标签: keras tensorflow2.0