广西民族大学高级人工智能课程—头歌实践教学实践平台-目标定位-CSDN博客

本文链接：https://blog.csdn.net/gxmzuai/article/details/135016539

代码文件

import os

import numpy as np  # linear algebra
np.random.seed(2022)
import pandas as pd  # data processing
from sklearn.model_selection import train_test_split

id_lookup_table = "facial-keypoints-detection/IdLookupTable.csv"
sample_Submission = "facial-keypoints-detection/SampleSubmission.csv"

# 读取数据集
train_df = pd.read_csv('training.csv')
idLookupTable = pd.read_csv(id_lookup_table)
sampleSumission = pd.read_csv(sample_Submission)

# 缺失值填充
train_df.fillna(method='ffill', inplace=True)

# 处理训练数据
image_df = train_df['Image']
imageArr = []
for i in range(0, len(image_df)):
    img = image_df[i].split()
    img = ['0' if x == '' else x for x in img]
    imageArr.append(img)
x_train = np.array(imageArr, dtype='float')
x_train = x_train.reshape(-1, 96, 96, 1)
print(x_train.shape)

keypoints_df = train_df.drop('Image', axis=1)
y_train = np.array(keypoints_df, dtype='float')
print(y_train.shape)

x_train, x_test, y_train, y_test = train_test_split(x_train, y_train)

from keras.models import Sequential
from keras.layers import Convolution2D, MaxPooling2D, BatchNormalization, Flatten, Dense, Dropout
from keras.layers.advanced_activations import LeakyReLU

# 构建模型
model = Sequential()
model.add(Convolution2D(32, (3, 3), padding='same', use_bias=False, input_shape=(96, 96, 1)))
model.add(LeakyReLU(alpha=0.1))
model.add(BatchNormalization())

model.add(Convolution2D(32, (3, 3), padding='same', use_bias=False))
model.add(LeakyReLU(alpha=0.1))
model.add(BatchNormalization())
model.add(MaxPooling2D(pool_size=(2, 2)))

########## Begin ##########
# 卷积层，卷积核数为 64，卷积核大小为 (3, 3)，padding 为 same，use_bias 设置为 False
model.add(Convolution2D(64, (3, 3), padding='same', use_bias=False))

# 激活函数层
model.add(LeakyReLU(alpha=0.1))

# 批标准化层
model.add(BatchNormalization())
########## End ##########

model.add(Convolution2D(64, (3, 3), padding='same', use_bias=False))
model.add(LeakyReLU(alpha=0.1))
model.add(BatchNormalization())
model.add(MaxPooling2D(pool_size=(2, 2)))

model.add(Convolution2D(96, (3, 3), padding='same', use_bias=False))
model.add(LeakyReLU(alpha=0.1))
model.add(BatchNormalization())

model.add(Convolution2D(96, (3, 3), padding='same', use_bias=False))
model.add(LeakyReLU(alpha=0.1))
model.add(BatchNormalization())
model.add(MaxPooling2D(pool_size=(2, 2)))
model.add(Convolution2D(128, (3, 3), padding='same', use_bias=False))
model.add(LeakyReLU(alpha=0.1))
model.add(BatchNormalization())

model.add(Convolution2D(128, (3, 3), padding='same', use_bias=False))
model.add(LeakyReLU(alpha=0.1))
model.add(BatchNormalization())
model.add(MaxPooling2D(pool_size=(2, 2)))

model.add(Convolution2D(256, (3, 3), padding='same', use_bias=False))
model.add(LeakyReLU(alpha=0.1))
model.add(BatchNormalization())

model.add(Convolution2D(256, (3, 3), padding='same', use_bias=False))
model.add(LeakyReLU(alpha=0.1))
model.add(BatchNormalization())
model.add(MaxPooling2D(pool_size=(2, 2)))

model.add(Convolution2D(512, (3, 3), padding='same', use_bias=False))
model.add(LeakyReLU(alpha=0.1))
model.add(BatchNormalization())

model.add(Convolution2D(512, (3, 3), padding='same', use_bias=False))
model.add(LeakyReLU(alpha=0.1))
model.add(BatchNormalization())

model.add(Flatten())
########## Begin ##########
# 全连接层，神经元数为 512 个，激活函数为 relu
model.add(Dense(512, activation='relu'))

# 随机将 10% 的神经元的失去活性
model.add(Dropout(0.1))

# 全连接层，神经元数为 30 个
model.add(Dense(30))

########## End ##########

model.summary()
# 加载模型
model.load_weights('model_weights.h5')

model.compile(optimizer='adam', loss='mean_squared_error', metrics=['mae', 'acc'])

# model.fit(x_train, y_train, batch_size=256, epochs=45, validation_split=2.0)
# model.save_weights('model_weights.h5')


########## Begin ##########
# 评估模型
score = model.evaluate(x_test, y_test, verbose=0)
print('Test score:', score[0])
print('Test accuracy:', score[1])
########## End ##########