klsfct / vgg16_02

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

此工程目前 强化学习,用的vgg16 模型 数据量

tfenv 3.6.8

Model: "sequential_1"


Layer (type) Output Shape Param #

conv2d_1 (Conv2D) (None, 224, 224, 64) 1792


activation_1 (Activation) (None, 224, 224, 64) 0


batch_normalization_1 (Batch (None, 224, 224, 64) 256


dropout_1 (Dropout) (None, 224, 224, 64) 0


conv2d_2 (Conv2D) (None, 224, 224, 64) 36928


activation_2 (Activation) (None, 224, 224, 64) 0


batch_normalization_2 (Batch (None, 224, 224, 64) 256


max_pooling2d_1 (MaxPooling2 (None, 112, 112, 64) 0


dense_1 (Dense) (None, 112, 112, 512) 33280


activation_3 (Activation) (None, 112, 112, 512) 0


batch_normalization_3 (Batch (None, 112, 112, 512) 2048


dropout_2 (Dropout) (None, 112, 112, 512) 0


dense_2 (Dense) (None, 112, 112, 2) 1026


activation_4 (Activation) (None, 112, 112, 2) 0

Total params: 75,586 Trainable params: 74,306 Non-trainable params: 1,280


C:/Users/Administrator/Desktop/耿霞/vgg16/train C:/Users/Administrator/Desktop/耿霞/vgg16/train/cracked_traindatavgg16_gaoerji.h5 Found 267 images belonging to 2 classes. done save the model

Process finished with exit code 0

About


Languages

Language:Python 100.0%