用神经网络实现人脸识别完整代码anaconda

以下是一个在 Anaconda 环境下使用 TensorFlow 实现的人脸识别代码，供参考：

1. 数据预处理部分：

import cv2
import os
import numpy as np

# 定义图片路径
img_path = "data/face_images/"
# 定义识别标签
label_dict = {"person1": 0, "person2": 1, "person3": 2}

# 定义函数进行数据预处理
def preprocess():
    # 存储预处理后的图像和标签
    images = []
    labels = []
    # 遍历文件夹中的每个文件
    for foldername in os.listdir(img_path):
        subject_path = os.path.join(img_path, foldername)
        for filename in os.listdir(subject_path):
            # 读取图像并进行灰度化和大小归一化处理
            img = cv2.imread(os.path.join(subject_path, filename), cv2.IMREAD_GRAYSCALE)
            img = cv2.resize(img, (100, 100))
            # 将图像数据和标签存储到数组中
            images.append(np.asarray(img, dtype=np.uint8))
            labels.append(label_dict[foldername])
    # 将标签转换为numpy数组
    labels = np.asarray(labels, dtype=np.int32)
    # 返回预处理后的数据
    return images, labels

# 进行数据预处理
images, labels = preprocess()

2. 模型训练部分：

import tensorflow as tf

# 定义模型架构
model = tf.keras.models.Sequential([
    tf.keras.layers.Flatten(input_shape=(100, 100)),
    tf.keras.layers.Dense(128, activation='relu'),
    tf.keras.layers.Dense(3)
])

# 编译模型
model.compile(optimizer='adam',
              loss=tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True),
              metrics=['accuracy'])

# 训练模型
model.fit(images, labels, epochs=10)

3. 模型测试部分：

import random

# 随机选择一个测试图像进行测试
test_img_path = "data/face_images/person1/1.jpg"
test_img = cv2.imread(test_img_path, cv2.IMREAD_GRAYSCALE)
test_img = cv2.resize(test_img, (100, 100))

# 对测试图像进行预测
prediction = model.predict(np.asarray(test_img, dtype=np.uint8).reshape(1, 100, 100))

# 输出预测结果
prediction_label = np.argmax(prediction)
for label, index in label_dict.items():
    if index == prediction_label:
        print("预测结果为：", label)
        break

4. 部署应用部分：

将训练好的模型保存为.h5文件，并在实际应用中加载模型文件进行人脸识别。具体实现代码可以根据实际需求进行编写。在 Anaconda 环境下，可以使用以下代码保存模型：

model.save("model.h5")

加载模型文件的代码如下：

import tensorflow as tf

# 加载模型
model = tf.keras.models.load_model("model.h5")

需要注意的是，在部署应用时需要考虑到数据隐私保护的问题，因此在实际应用中需要遵循相关法律法规和隐私保护原则。