写一个识别猫和狗的神经网络算法代码
时间: 2023-03-14 13:11:52 浏览: 72
我们可以使用深度学习神经网络来设计一个识别猫和狗的算法代码,如下所示:from keras.models import Sequential
from keras.layers import Convolution2D
from keras.layers import MaxPooling2D
from keras.layers import Flatten
from keras.layers import Dense#初始化CNN
classifier = Sequential()
#卷积层
classifier.add(Convolution2D(32, 3, 3, input_shape = (64, 64, 3), activation = 'relu'))
#池化层
classifier.add(MaxPooling2D(pool_size = (2, 2)))
#卷积层
classifier.add(Convolution2D(32, 3, 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'))
#编译CNN
classifier.compile(optimizer = 'adam', loss = 'binary_crossentropy', metrics = ['accuracy'])
#训练模型
classifier.fit_generator(train_set,
samples_per_epoch = 8000,
nb_epoch = 25,
validation_data = test_set,
nb_val_samples = 2000)