CNN.py 文件源码

def getNN(n):
    """
    ?????????
    ???????????VGG???
    """
    nn=Sequential()
    nn.add(Convolution2D(32,(3,3),input_shape=(30,30,1),activation='relu'))
    nn.add(MaxPooling2D(pool_size=(2, 2)))
    nn.add(Convolution2D(16,(3,3),activation='relu'))
    nn.add(Dropout(0.2))
    nn.add(Convolution2D(8,(3,3),activation='relu'))
    nn.add(MaxPooling2D(pool_size=(2, 2)))
    nn.add(Convolution2D(8,(3,3),activation='relu'))
    nn.add(Dense(50,activation='tanh'))
    nn.add(Dropout(0.2))
    nn.add(Dense(50,activation='tanh'))
    nn.add(Flatten())
    nn.add(Dense(n,activation='sigmoid'))
    nn.compile(optimizer='rmsprop',loss='categorical_crossentropy')
    return nn