基于OpenCV的视频人脸检测程序问题求助:人脸截取异常与帧图片保存不全
解决方案:修复人脸检测样本保存的两个问题
咱们先拆解你遇到的两个问题,再给出修改后的代码:
问题1:保存整帧而非人脸区域
你当前代码里cv2.imwrite("../output_model/videos/image.jpg", image)保存的是处理后的整帧画面(也就是画了框、模糊了人脸的原视频帧),而不是你用image2 = image[y:(y+h),x:(x+w)]截取的人脸区域。要解决这个问题,需要把保存的对象换成截取到的人脸区域。
问题2:仅保存1张图片
因为你每次都用同一个文件名image.jpg保存,新的图片会直接覆盖旧的,所以最后只留下一张。解决方案是给每张人脸样本生成唯一的文件名,比如用计数器递增命名,就能避免覆盖问题。
修改后的完整代码
import cv2 model = cv2.CascadeClassifier("../model/haarcascade_frontalface_alt2.xml") cap = cv2.VideoCapture('../video/videoplayback.mp4') # 加载视频文件 face_count = 0 # 人脸计数器,用于生成唯一文件名 while True: ret, image = cap.read() # 检查是否成功读取帧,防止视频结束后报错 if not ret: break gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY) bounding_box = model.detectMultiScale(gray, scaleFactor=1.01, minNeighbors=5, minSize=(30, 30), flags=cv2.CASCADE_SCALE_IMAGE) for (x, y, w, h) in bounding_box: # 截取人脸区域 face_img = image[y:y+h, x:x+w] # 生成唯一文件名,比如face_0.jpg、face_1.jpg... save_path = f"../output_model/videos/face_{face_count}.jpg" # 保存人脸样本 cv2.imwrite(save_path, face_img) face_count += 1 # 计数器递增 # 保留你原本的画框和模糊人脸的代码,用于实时预览效果 cv2.rectangle(image, (x, y), (x + w, y + h), (0, 255, 0), 2) blurred_face = cv2.blur(face_img, (40,40)) image[y:y+h, x:x+w] = blurred_face cv2.imshow("hasil", image) if cv2.waitKey(1) & 0xFF == ord('q'): break cap.release() cv2.destroyAllWindows()
关键修改点说明
- 添加
face_count计数器,每次保存人脸时自动递增,确保每个样本文件名唯一,不会被覆盖。 - 将
cv2.imwrite的保存对象从整帧image换成了截取的人脸区域face_img。 - 增加
if not ret: break判断,避免视频读取完毕后继续执行代码导致报错。 - 调整了变量名(比如
image2改成face_img),让代码更易读。
这样修改后,程序就能每帧检测到人脸就保存对应的人脸样本,且不会覆盖之前的文件啦!
内容的提问来源于stack exchange,提问作者alexander scream
相关产品推荐
相关产品推荐

