使用DeepFace提取含人脸视频帧遇ValueError问题,求解决方案
解决DeepFace检测视频人脸帧时的报错与无效存储问题
问题分析
核心矛盾点:
- 默认
enforce_detection=True时,DeepFace检测不到人脸会直接抛出ValueError,中断程序 - 设置
enforce_detection=False后,无法区分帧中是否存在人脸,导致所有帧都被存入数组
解决方案
通过异常捕获机制处理无脸帧的报错,仅保留成功检测到人脸的帧。以下是修改后的完整代码:
# Import Libraries from deepface import DeepFace import cv2 # Path of the video video_file_path = '/content/drive/My Drive/Colab Notebooks/FYP Project/Data Preprocessing/youtube_clip_001.mp4' # Reading the video vidcap = cv2.VideoCapture(video_file_path) # Extracting the frames frames = [] while True: ret, frame = vidcap.read() if not ret: break try: # 尝试检测人脸,使用默认的enforce_detection=True DeepFace.detectFace(frame) # 执行到此处说明检测到人脸,存入数组 frames.append(frame) except ValueError: # 捕获无脸帧的报错,直接跳过该帧 continue # 释放视频资源 vidcap.release()
补充优化方案
如果需要同时获取人脸的位置信息,可以改用DeepFace.extract_faces方法,它会返回包含人脸区域的列表,逻辑更直观:
# 替换原检测部分的代码 try: faces = DeepFace.extract_faces(frame, enforce_detection=True) if len(faces) > 0: frames.append(frame) except ValueError: continue
该方法检测到人脸时返回非空列表,无脸时抛出ValueError,同样通过异常捕获跳过无效帧;若设置enforce_detection=False,无脸时会返回空列表,可直接通过len(faces) > 0判断是否保留帧。
内容的提问来源于stack exchange,提问作者Pawara Siriwardhane
相关产品推荐
相关产品推荐

