此工程目前 强化学习,用的vgg16 模型 数据量
tfenv 3.6.8
Model: "sequential_1"
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
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