帮我写一个使用基于深度学习的TensorFlow模型检测其中的人脸的代码
时间: 2024-02-19 14:04:13 浏览: 61
基于TensorFlow人脸识别源代码
5星 · 资源好评率100%
好的,这里给您提供一个简单的 TensorFlow 人脸检测模型代码:
```python
import tensorflow as tf
import cv2
# 加载模型,这里使用的是基于 MTCNN 的人脸检测模型
model = tf.keras.models.load_model('path/to/model.h5')
# 加载待检测的图像
img = cv2.imread('path/to/image.jpg')
# 调整图像大小
img = cv2.resize(img, (224, 224))
# 归一化图像
img = img / 255.0
# 将图像转换为 4D 张量
img = tf.expand_dims(img, axis=0)
# 进行预测
preds = model.predict(img)
# 取出预测结果
boxes = preds[0]
# 绘制人脸框
for box in boxes:
x1, y1, x2, y2 = box
x1, y1, x2, y2 = int(x1), int(y1), int(x2), int(y2)
cv2.rectangle(img, (x1, y1), (x2, y2), (0, 255, 0), 2)
# 显示结果图像
cv2.imshow('result', img)
cv2.waitKey(0)
cv2.destroyAllWindows()
```
需要注意的是,这里使用的是基于 MTCNN 的人脸检测模型,您需要先下载并安装 MTCNN 模型才能运行上述代码。
阅读全文