加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
model.py 648 Bytes
一键复制 编辑 原始数据 按行查看 历史
Arctanxy 提交于 2018-09-16 22:48 . train_lfw
'''
建立两个conv network,然后将两个network的结果输入dicision network中
'''
import tensorflow as tf
from tensorflow import keras
from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Conv2D,MaxPool2D,Dropout,Dense,Flatten
model = Sequential()
model.add(Conv2D(32,kernel_size=(3,3),activation='relu'))
model.add(Conv2D(64,(3,3),activation='relu'))
model.add(MaxPool2D(pool_size=(2,2)))
model.add(Dropout(0.25))
model.add(Flatten())
model.add(Dense(1,activation='sigmoid'))
model.compile(loss=keras.losses.MSE,
optimizer=keras.optimizers.Adam(),
metrics=['accuracy'])
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化