python小项目 | python 百度方案批量进行人脸识别（图片/视频）

最新推荐文章于 2025-07-08 17:40:34 发布

原创

最新推荐文章于 2025-07-08 17:40:34 发布 · 784 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#python #音视频 #开发语言

本文介绍了如何使用Python和PaddleHub库中的pyramidbox_lite_server模块实现一个界面化的工具，该工具能自动处理包含图片和视频的文件夹，进行人脸识别并截取人脸图片。同时，文章详细展示了如何读取特殊字符路径的图片和处理不同类型的视频帧。

python 人脸识别并截取人脸保存

实现的功能

1.界面化，拖入包含图片/视频的文件夹后，一键统一处理
2.实现人脸识别并截取保存功能
3.可以处理图片和视频

人脸识别并提取

import os, time, sys, cv2, random
import tkinter as tk
import windnd
import numpy as np

baseDirPath = sys.path[0]   #在导入paddlehub之前执行， 否则地址会被改变
print(baseDirPath)

import paddlehub as hub  #百度方案

#加载模型
module = hub.Module(name = 'pyramidbox_lite_server')

VideoType = ['mp4', 'flv', 'webm', 'wmv', 'ts']
PicType = ['jpg', 'png', 'bmp', 'jpeg']  

video_path = baseDirPath + '\\01\\movie'
face_pic_dir = baseDirPath + '\\01\\pic'
  
if not os.path.exists(face_pic_dir):
  os.makedirs(face_pic_dir)

###########################################################
#读含中文/韩文/日文等特殊字符路径的图片
def cv_imread(in_path):
  im = cv2.imdecode(np.fromfile(in_path, dtype=np.uint8),-1)
  return im
  
#路径中有中文名 cv2写
def cv_imwrite(out_path, imp_np):
  imp_type = '.' + out_path.split('.')[-1]
  cv2.imencode(imp_type, imp_np)[1].tofile(out_path)
############################################################

###############################################
#1.pic--face   图片人脸识别
def getFaceFromPic(vin_path, vout_dir):
  src = cv_imread(vin_path)
  pic_name = vin_path.split('\\')[-

最低0.47元/天解锁文章