Блог

Я построил модель keras для классификации изображений, в наборе данных есть 5 классов, но когда модель классифицирует изображение, я обнаружил, что около 200 изображений не классифицируются, может ли кто-нибудь помочь мне, как я могу найти те изображения, которые не классифицируются.Это моя модель

     img_width= img_rows
    img_height = img_cols
    classes_num = 5
    epochs = 10

    model =Sequential()

    model.add(Conv2D(32,(3,3), input_shape=(img_width, img_height, 3)))
    model.add(Activation('relu'))
    model.add(MaxPooling2D(pool_size=(2,2)))

    model.add(Conv2D(32,(3,3), input_shape=(img_width, img_height, 3)))
    model.add(Activation('relu'))
    model.add(MaxPooling2D(pool_size=(2,2)))

    model.add(Conv2D(64,(3,3), input_shape=(img_width, img_height, 3)))
    model.add(Activation('relu'))
    model.add(MaxPooling2D(pool_size=(2,2)))

    model.add(Flatten())
    model.add(Dense(256))
    model.add(Activation('relu'))
    model.add(Dropout(0.1))
    model.add(Dense(classes_num ))
    model.add(Activation('sigmoid'))

    model.compile(optimizer='adam', loss='binary_crossentropy',
                            metrics=['accuracy',mean_pred,recall,precision, fmeasure,
                                     matthews_correlation,kullback_leibler_divergence,
                                     binary_crossentropy])
    model.save('model.h5')
    model.summary()
    print('model complied!!')

    print('starting training....')
    history = model.fit(X_train, Y_train, epochs=epochs, batch_size=64,validation_data=(X_test, Y_test))
  

»’