Выясните, что изображение не соответствует классификации при использовании модели keras в качестве классификатора изображения. - PullRequest
0 голосов
/ 21 апреля 2019

Я построил модель 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))

'' '

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...