keras实现图像识别—cat

目的：识别图片中是否有猫
训练集x_train（209,64,64,3）,y_train(209,)
测试集x_test（50,64,64,3），y_test(50,)

设计的神经网络结构图如下：
卷积层——池化层——卷积层——池化层——卷积层——池化层——平铺展开——全连接层——全连接层——全连接层

获取数据

# coding: utf-8
# 2019/7/28 19:23
import h5py
import numpy as np
def get_train():
    f = h5py.File('dataset/train_catvnoncat.h5','r')
    x_train = np.array(f['train_set_x'])#训练集数据  将数据转化为np.array
    y_train = np.array(f['train_set_y'])#训练集标签
    return x_train,y_train
def get_test():
    f = h5py.File('dataset/test_catvnoncat.h5', 'r')
    x_test = np.array(f['test_set_x'])#测试集数据  将数据转化为np.array
    y_test = np.array(f['test_set_y'])#测试集标签
    return x_test,y_test

def preprocess(X):
    #将X标准化，从0-255变成0-1
    X =X / 255
    return X


if __name__ == '__main__':
    x1,y1 = get_train()
    x2,y2 = get_test()
    print(x1.shape,y1.shape)
    print(x2.shape,y2.shape