Вывод Keras Predict_generator каждый раз отличается

за последние 2 месяца я застрял с этой проблемой, и это сводило меня с ума, пока я не понял, что мой вектор «вероятностей» из предсказать_генератор просто неверен.

Я использую keras 2, и у меня есть папка test с подкаталогами, содержащими изображения (не обязательно одинаковое количество изображений).

затем я импортирую свою модель, загружаю веса и делаю это:

from keras.applications import ResNet50
model = ResNet50(include_top=True, weights=None, input_shape=(3,224,224),classes=N)
model.load_weights(model_path)
probs1 = model.predict_generator(batches, steps=batches.n/64, verbose=1)
probs2 = model.predict_generator(batches, steps=batches.n/64, verbose=1)

и я не знаю, почему, но probs1 != probs2, когда probs2 кажется «правильным» прогнозом.

P.S. batches.n/64 не является целым числом

Что я должен делать?


person Jenia Golbstein    schedule 22.02.2018    source источник


Ответы (1)


Взгляните на эту ветку. Но это уже должно быть исправлено. Попробуйте поставить datagen.reset() перед model.predict_generator().

person Peter    schedule 16.04.2018