Сверточная нейронная сеть не запускает эфос - PullRequest
0 голосов
/ 06 апреля 2020

Я просто строю CNN, чтобы классифицировать изображения по двум категориям, коды не дают мне никаких ошибок при запуске. Когда я запускаю программу, она запускается, но не показывает эпохи. Вот мой код

from keras.layers.convolutional import Convolution2D
from keras.layers.convolutional import MaxPooling2D
from keras.models import Sequential
from keras.layers import Flatten
from keras.layers import Dense

classifier=Sequential()
classifier.add(Convolution2D(32,3,3,input_shape=(64,64,3),activation='relu'))  
classifier.add(MaxPooling2D(pool_size=(2,2)))
classifier.add(Flatten())

classifier.add(Dense(output_dim=128,activation='relu'))
classifier.add(Dense(output_dim=1,activation='sigmoid'))
classifier.compile(optimizer='adam',loss='binary_crossentropy',metrics=['accuracy'])

from keras.preprocessing.image import ImageDataGenerator

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

test_datagen = ImageDataGenerator(rescale=1./255)

training_set = train_datagen.flow_from_directory(
    'dataset/training_set',
    target_size=(64, 64),
    batch_size=32,
    class_mode='binary')

test_set = test_datagen.flow_from_directory(
    'dataset/test_set',
    target_size=(64, 64),
    batch_size=32,
    class_mode='binary')

classifier.fit_generator( 
        training_set,
        steps_per_epoch=8000,
        epochs=25,
        validation_data=test_set,
        verbose=2,
        validation_steps=2000 )

После запуска этого кода он не отображает эпохи, он показывает только эпоху 1/25, и все. это изображение вывода the output of the program, it only shows this

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