如何解决收到错误“ IndexError:列表索引超出范围”
有我的代码
model = Sequential()
model.add(Conv2D(32,(3,3),activation='relu',input_shape=(48,120,1)))
model.add(Conv2D(64,activation='relu'))
model.add(Dropout(0.25))
model.add(MaxPooling2D((2,2)))
model.add(Conv2D(128,2)))
model.add(Flatten())
model.add(Dense(512,activation='relu'))
model.add(Dense(62*4,activation='softmax'))
# print(model.summary())
model.compile(
loss='categorical_crossentropy',optimizer=RMSprop(lr=1e-4),metrics=['accuracy']
)
history = model.fit_generator(
DataGenerator(train_dir,(48,1)),steps_per_epoch=32,epochs=30,validation_data=DataGenerator(test_dir,validation_steps=32
)
DataGenerator
class DataGenerator(Sequence):
def __init__(self,image_path,shape,to_fit=True,batch_size=32,):
self.image_path = image_path
self.shape = shape
self.to_fit = to_fit
self.batch_size = batch_size
self.all_images = self.get_all_images()
def __len__(self):
# return the number of batches per epoch
return len(os.listdir(self.image_path))
def __getitem__(self,index):
# return a batch of images and (labels) if we are predicting
if self.to_fit:
return self._genX_(index),self._genY_(index)
else:
return self._genX_(index)
def _genX_(self,index):
X = self.all_images[index*self.batch_size:(index+1)*self.batch_size]
X = [cv2.imread(img,0) for img in X]
X = [img.reshape(self.shape) for img in X]
X = [img.astype('float32')/255 for img in X]
return X
def _genY_(self,index):
y = self.all_images[index*self.batch_size:(index+1)*self.batch_size]
return y
def on_epoch_end(self):
# do something after every epoch
return
def get_all_images(self):
return os.listdir(self.image_path)
我有什么
Epoch 1/30
Traceback (most recent call last):
File ".\test_1.py",line 42,in <module>
validation_steps=32
File "E:\Environments\Python\python37\lib\site-packages\keras\legacy\interfaces.py",line 91,in wrapper
return func(*args,**kwargs)
File "E:\Environments\Python\python37\lib\site-packages\keras\engine\training.py",line 1732,in fit_generator
initial_epoch=initial_epoch)
File "E:\Environments\Python\python37\lib\site-packages\keras\engine\training_generator.py",line 220,in fit_generator
reset_metrics=False)
File "E:\Environments\Python\python37\lib\site-packages\keras\engine\training.py",line 1508,in train_on_batch
class_weight=class_weight)
File "E:\Environments\Python\python37\lib\site-packages\keras\engine\training.py",line 579,in _standardize_user_data
exception_prefix='input')
File "E:\Environments\Python\python37\lib\site-packages\keras\engine\training_utils.py",line 87,in standardize_input_data
if isinstance(data[0],list):
IndexError: list index out of range
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。