python 二分类的实例_keras分类之二分类实例(Cat and dog)

1. 数据准备

在文件夹下分别建立训练目录train，验证目录validation，测试目录test，每个目录下建立dogs和cats两个目录，在dogs和cats目录下分别放入拍摄的狗和猫的图片，图片的大小可以不一样。

2. 数据读取

# 存储数据集的目录

base_dir = 'E:/python learn/dog_and_cat/data/'

# 训练、验证数据集的目录

train_dir = os.path.join(base_dir, 'train')

validation_dir = os.path.join(base_dir, 'validation')

test_dir = os.path.join(base_dir, 'test')

# 猫训练图片所在目录

train_cats_dir = os.path.join(train_dir, 'cats')

# 狗训练图片所在目录

train_dogs_dir = os.path.join(train_dir, 'dogs')

# 猫验证图片所在目录

validation_cats_dir = os.path.join(validation_dir, 'cats')

# 狗验证数据集所在目录

validation_dogs_dir = os.path.join(validation_dir, 'dogs')

print('total training cat images:', len(os.listdir(train_cats_dir)))

print('total training dog images:', len(os.listdir(train_dogs_dir)))

print('total validation cat images:', len(os.listdir(validation_cats_dir)))

print('total validation dog images:', len(os.listdir(validation_dogs_dir)))

3. 模型建立

# 搭建模型

model = Sequential()

model.add(Conv2D(32, (3, 3), activation='relu',

input_shape=(150, 150, 3)))

model.add(MaxPooling2D((2, 2)))

model.add(Conv2D(64, (3, 3), activation='relu'))

model.add(MaxPooling2D((2, 2)))

model.add(Conv2D(128, (3, 3), activation='relu'))

model.add(MaxPooling2D((2, 2)))

model.add(Conv2D(128, (3, 3), activation='relu'))

model.add(MaxPooling2D((2, 2)))

model.add(Flatten())

model.add(Dense(512, activation='relu'))

model.add(Dense(1, activation='sigmoid'))

print(model.summary())

model.compile(loss='binary_crossentropy',

optimizer=RMSprop(lr=1