CNN人脸识别,代码
时间: 2023-11-09 11:57:57 浏览: 61
对于人脸识别,卷积神经网络(Convolutional Neural Network,CNN)是一种常用的方法。下面是一个简单的CNN人脸识别的代码示例(使用Python和TensorFlow):
```python
import tensorflow as tf
# 创建CNN模型
def create_model():
model = tf.keras.models.Sequential([
tf.keras.layers.Conv2D(32, (3, 3), activation='relu', input_shape=(64, 64, 3)),
tf.keras.layers.MaxPooling2D((2, 2)),
tf.keras.layers.Conv2D(64, (3, 3), activation='relu'),
tf.keras.layers.MaxPooling2D((2, 2)),
tf.keras.layers.Conv2D(64, (3, 3), activation='relu'),
tf.keras.layers.Flatten(),
tf.keras.layers.Dense(64, activation='relu'),
tf.keras.layers.Dense(10, activation='softmax')
])
return model
# 加载训练数据和标签
train_images = ...
train_labels = ...
# 创建模型
model = create_model()
# 编译模型
model.compile(optimizer='adam',
loss=tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True),
metrics=['accuracy'])
# 训练模型
model.fit(train_images, train_labels, epochs=10)
# 使用模型进行预测
test_images = ...
predictions = model.predict(test_images)
# 输出预测结果
print(predictions)
```
以上代码只是一个简单的示例,实际应用中可能需要更复杂的模型结构和数据预处理步骤。你可以根据自己的需求进行修改和扩展。