【问题标题】:Exception: Error when checking model target: expected dense_3 to have shape (None, 1000) but got array with shape (32, 2)异常:检查模型目标时出错:预期 dense_3 具有形状 (None, 1000) 但得到的数组具有形状 (32, 2)
【发布时间】:2017-01-13 02:38:41
【问题描述】:

如何为我的数据创建 VGG-16 序列?

数据有以下内容:

model = Sequential() 
model.add(ZeroPadding2D((1, 1), input_shape=(3, img_width, img_height))) model.add(Convolution2D(64, 3, 3, activation='relu', name='conv1_1')) model.add(ZeroPadding2D((1, 1)))
model.add(Convolution2D(64, 3, 3, activation='relu', name='conv1_2'))
model.add(MaxPooling2D((2, 2), strides=(2, 2)))

model.add(ZeroPadding2D((1, 1))) 
model.add(Convolution2D(128, 3, 3, activation='relu', name='conv2_1')) model.add(ZeroPadding2D((1, 1)))
model.add(Convolution2D(128, 3, 3, activation='relu', name='conv2_2'))
model.add(MaxPooling2D((2, 2), strides=(2, 2)))

model.add(ZeroPadding2D((1, 1))) 
model.add(Convolution2D(256, 3, 3, activation='relu', name='conv3_1')) model.add(ZeroPadding2D((1, 1)))
model.add(Convolution2D(256, 3, 3, activation='relu', name='conv3_2'))
model.add(ZeroPadding2D((1, 1))) 
model.add(Convolution2D(256, 3, 3, activation='relu', name='conv3_3')) model.add(MaxPooling2D((2, 2), strides=(2, 2)))

model.add(ZeroPadding2D((1, 1))) 
model.add(Convolution2D(512, 3, 3, activation='relu', name='conv4_1')) model.add(ZeroPadding2D((1, 1)))
model.add(Convolution2D(512, 3, 3, activation='relu', name='conv4_2'))
model.add(ZeroPadding2D((1, 1))) 
model.add(Convolution2D(512, 3, 3, activation='relu', name='conv4_3')) model.add(MaxPooling2D((2, 2), strides=(2, 2)))

model.add(ZeroPadding2D((1, 1))) 
model.add(Convolution2D(512, 3, 3, activation='relu', name='conv5_1')) model.add(ZeroPadding2D((1, 1)))
model.add(Convolution2D(512, 3, 3, activation='relu', name='conv5_2'))
model.add(ZeroPadding2D((1, 1))) 
model.add(Convolution2D(512, 3, 3, activation='relu', name='conv5_3')) model.add(MaxPooling2D((2, 2), strides=(2, 2)))

model.add(Flatten()) 
model.add(Dense(4096, activation='relu'))
model.add(Dropout(0.5)) 
model.add(Dense(4096, activation='relu'))
model.add(Dropout(0.5)) 
model.add(Dense(1000, activation='softmax'))

sgd = SGD(lr=0.1, decay=1e-6, momentum=0.9, nesterov=True)
model.compile(optimizer=sgd, loss='categorical_crossentropy')

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

test_datagen = ImageDataGenerator(rescale=1./255)

train_generator = train_datagen.flow_from_directory(
        train_data_dir,
        target_size=(img_width, img_height),
        batch_size=32)

validation_generator = test_datagen.flow_from_directory(
        validation_data_dir,
        target_size=(img_width, img_height),
        batch_size=32)

model.fit_generator(
        train_generator,
        samples_per_epoch=2000,
        nb_epoch=1,
        verbose=1,
        validation_data=validation_generator,
        nb_val_samples=800)

json_string = model.to_json()  
open('my_model_architecture.json','w').write(json_string) 
model.save_weights('Second_try.h5')

我遇到了一个错误:

异常:检查模型目标时出错:预期的 dense_3 有 形状 (None, 32) 但得到了形状 (32, 2) 的数组

如何更改Dense 以使其正常工作?

【问题讨论】:

    标签: keras


    【解决方案1】:

    我有 10 种,
    我已经通过
    解决了这个问题 变化:

    model.add(Dense(1000, activation='softmax'))
    

    到:

    model.add(Dense(10, activation='softmax'))
    

    然后就可以了。

    【讨论】:

    • 10种是指10类吗?
    【解决方案2】:

    这里应该有类的总数而不是 1000,因为它是输出层。

    model.add(Dense(1000, activation='softmax')) 
    

    标签的形状(或 Y_train/Y_test)也应该是(类总数,记录总数)。

    这帮助我解决了类似的错误。

    【讨论】:

      猜你喜欢
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 2020-05-19
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 2018-07-29
      • 1970-01-01
      相关资源
      最近更新 更多