Question

目前正在学习深度学习大约 2-3 天，并在我尝试训练数据时出现此错误

<块引用>

ValueError: 要求检索元素 0，但序列长度为 0

google collab 说错误在这里

<块引用>

 56     validation_steps=totalVal // batch_size,
 57     verbose=1,

---> 58 个回调=[tensorboard])

我在这里有如下代码来制作我需要的模型

代码

import os
from imutils import paths
from keras.preprocessing.image import ImageDataGenerator
from keras import backend as K
from keras.callbacks import TensorBoard
from time import time
from keras.models import Sequential
from keras.layers import Conv2D, MaxPooling2D
from keras.layers import Activation, Dropout, Flatten, Dense

IM_WIDTH=224
IM_HEIGHT=224
EPOCH=2
batch_size=32

ORIG_INPUT_DATASET="/content/drive/MyDrive/DataFix/DataJadi"
BASE_PATH="/content/drive/MyDrive/DataFix/DataJadi/"

TRAIN_PATH = os.path.sep.join([BASE_PATH, "training"])
VAL_PATH = os.path.sep.join([BASE_PATH, "validation"])
TEST_PATH = os.path.sep.join([BASE_PATH, "testing"])

totalTrain = len(list(paths.list_images(TRAIN_PATH)))
totalVal = len(list(paths.list_images(VAL_PATH)))

def ClassicalModel(input_shape):
  model = Sequential()
  model.add(Conv2D(32, (3, 3), input_shape=input_shape))
  model.add(Activation('relu'))
  model.add(MaxPooling2D(pool_size=(2, 2)))
  model.add(Conv2D(32, (3, 3)))
  model.add(Activation('relu'))
  model.add(MaxPooling2D(pool_size=(2, 2)))
  model.add(Conv2D(64, (3, 3)))
  model.add(Activation('relu'))
  model.add(MaxPooling2D(pool_size=(2, 2)))
  model.add(Flatten())
  model.add(Dense(64,activation='relu'))
  model.add(Dropout(0.5))
  model.add(Dense(2, activation='softmax'))
  model.compile(optimizer='adam',
                loss='categorical_crossentropy', metrics=['accuracy'])
  
  return model

train_datagen = ImageDataGenerator(
    rescale=1 / 255,
    shear_range=0.2,
    zoom_range=0.2,
    horizontal_flip=True
)

val_datagen = ImageDataGenerator(rescale=1 / 255)

train_generator = train_datagen.flow_from_directory(TRAIN_PATH,
                                                    target_size=(IM_WIDTH, IM_HEIGHT),
                                                    batch_size=batch_size,
                                                    class_mode='categorical')

validation_generator = val_datagen.flow_from_directory(
    VAL_PATH,
    target_size=(IM_WIDTH, IM_HEIGHT),
    batch_size=batch_size,
    class_mode='categorical'
)
tensorboard=TensorBoard(log_dir="logs/{}".format(time()))
model = ClassicalModel(input_shape=(batch_size, IM_WIDTH, IM_HEIGHT))
model.fit_generator(
    train_generator,
    steps_per_epoch=totalTrain // batch_size,
    epochs=EPOCH,
    validation_data=validation_generator,
    validation_steps=totalVal // batch_size,
    verbose=1,
    callbacks=[tensorboard])
model.save_weights('GerakanUjian.h5')
model.save('GerakanUjian.h5')

直到现在仍然没有找到解决方案以及如何解决它:( 任何答案将不胜感激非常感谢你们

ValueError：要求检索元素 0，但序列长度为 0

0 个答案: