ValueError：数据基数不明确确保所有数组包含相同数量的样本？

问题描述

我正在尝试使用 MobileNet 模型，但面临上述问题。我不知道是不是 由于 train_test_split 或 else 发生。架构如下图 我可以在这里使用 model.fit 而不是 model.fit_generator 吗？

    mobilenet = MobileNet(input_shape=(224,224,3),weights='imagenet',include_top=False)
    
    # don't train existing weights
    for layer in mobilenet.layers:
      layer.trainable = False
     
    folders = glob('/content/drive/MyDrive/AllClasses/*')
    print("Total number of classes are",len(folders))  

    x = Flatten()(mobilenet.output)
    
    prediction = Dense(len(folders),activation='softmax')(x)
    
    model = Model(inputs=mobilenet.input,outputs=prediction)
    
    model.summary()
    
    model.compile(loss='categorical_crossentropy',optimizer='adam',metrics=['accuracy'])
    
  
  dataset = ImageDataGenerator(rescale=1./255)
    
    dataset = dataset.flow_from_directory('/content/drive/MyDrive/AllClasses',target_size=(224,224),batch_size=32,class_mode='categorical',color_mode='grayscale')

    train_data,test_data = train_test_split(dataset,random_state=42,test_size=0.20,shuffle=True)
 
    r = model.fit(train_data,validation_data=(test_data),epochs=5)

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）