基于Openface在ubuntu上抽取人脸图像

如题目所示,直接上代码

NOTE:代码中的路径注意更换

python 复制代码
import os
import glob
import time
import shutil
import subprocess
import concurrent.futures
from joblib import Parallel, delayed
from tqdm import tqdm

#------------------------------------------#
# 对单个视频文件进行处理
# 适配于Ubuntu-20.04操作系统
#------------------------------------------#
def process_one_video(video_file, in_dir, out_dir, img_size=112):
    file_name = os.path.splitext(os.path.basename(video_file))[0]
    out_dir   = os.path.join(out_dir, file_name)

    if os.path.exists(out_dir):
        print(f'Note: "{out_dir}" already exist!')
        return video_file
    else:
        os.makedirs(out_dir)

    cmd = f'/home/data/openface/OpenFace/build/bin/FeatureExtraction -f "{video_file}" -out_dir "{out_dir}" -simalign -simsize {img_size} -format_aligned jpg -nomask'
    subprocess.call(cmd, shell=True)

    return video_file

#----------------------------------------#
# 基于上面的process_one_video函数构建主函数
# 默认使用多进程进行处理
#----------------------------------------#
def main(video_dir, out_dir, multi_process=True, video_template_path='*.mp4', img_size=112):
    #---------------------------------#
    # 根据视频的类型不同 需要更换后缀名
    #---------------------------------#
    video_files = glob.glob(os.path.join(video_dir, video_template_path))  # with emo dir

    n_files     = len(video_files)
    print(f'Total videos: {n_files}.')

    if not os.path.exists(out_dir):
        os.makedirs(out_dir)
        
    start_time = time.time()
    #-------------------------#
    # 多进程处理
    # 我们默认进行多线程处理
    #-------------------------#
    if multi_process:
        Parallel(n_jobs=256)(delayed(process_one_video)(video_file, video_dir, out_dir, img_size) for video_file in tqdm(video_files))
    else:
        for i, video_file in enumerate(video_files, 1):
            print(f'Processing "{os.path.basename(video_file)}"...')
            process_one_video(video_file, video_dir, out_dir, img_size)
            print(f'"{os.path.basename(video_file)}" done, rate of progress: {100.0 * i / n_files:3.0f}% ({i}/{n_files})')

    end_time = time.time()
    print('Time used for video face extraction: {:.1f} s'.format(end_time - start_time))
 
#------------------------#
# 对目录的文件结构进行复制
#------------------------#
def copy_one_video(src_dir, tgt_dir):
    shutil.copytree(src_dir, tgt_dir)
    print(f'Copy "{src_dir}" to "{tgt_dir}"')


if __name__ == '__main__':
    #--------------------------------#
    # 确定数据集根目录与视频地址
    #--------------------------------#
    video_dir    = '/home/data/labeled/data/clips'
    img_size     = 256

    #-----------------------------#
    # 确定视频模板与输出文件夹地址
    #-----------------------------#
    file_ext            = 'mp4'
    video_template_path = f'*.{file_ext}'
    out_dir             = "/home/data/labeled/data/openface" 

    #----------------------------------#
    # 1. 使用openface进行人脸的抽取
    #----------------------------------#
    main(video_dir, out_dir, video_template_path=video_template_path, multi_process=False, img_size=img_size)

    #------------------------------#
    # 2. 重新组织已提取人脸的文件夹
    # P.S. 最后输出为face_aligned
    #------------------------------#
    src_root = out_dir
    tgt_root = out_dir.replace('openface', 'face_aligned')
    count    = 0
    src_dirs, tgt_dirs = [], []

    for sample_dir in os.scandir(src_root):
            sample_name = sample_dir.name

            tgt_dir     = os.path.join(tgt_root, sample_name) # organize videos in the original way
            src_dir     = os.path.join(sample_dir, f'{sample_name}_aligned')

            src_dirs.append(src_dir)
            tgt_dirs.append(tgt_dir)
            count += 1

    print(f'Total videos: {count}.')
    Parallel(n_jobs=64)(delayed(copy_one_video)(src_dir, tgt_dir) for src_dir, tgt_dir in tqdm(zip(src_dirs, tgt_dirs)))

openface的安装可以参考我的另外一篇blog:
https://blog.csdn.net/m0_47623548/article/details/138171121?spm=1001.2014.3001.5501

相关推荐
dualven_in_csdn1 小时前
搞了两天的win7批处理脚本问题
java·linux·前端
思通数科大数据舆情2 小时前
工业安全零事故的智能守护者:一体化AI智能安防平台
人工智能·安全·目标检测·计算机视觉·目标跟踪·数据挖掘·知识图谱
晨曦backend2 小时前
Vim 匹配跳转与搜索命令完整学习笔记
linux·编辑器·vim
bing_feilong4 小时前
Windows上SSH连接Ubuntu失败
windows·ubuntu·ssh
爬呀爬的水滴4 小时前
解决Ubuntu24.04版本,右键没有共享选项的问题
linux·服务器·ubuntu·samba·共享文件夹
IT coke4 小时前
centos7部署AWStats日志分析系统
linux·运维·centos
1296004525 小时前
机器学习的可解释性
人工智能·深度学习·自然语言处理·transformer
雾岛心情5 小时前
【黑客与安全】Linux的常用命令之系统架构信息获取系列命令
linux·运维·服务器
杯莫停丶5 小时前
Linux基础指令大全
linux·运维·chrome
MYH5165 小时前
神经元激活函数在神经网络里起着关键作用
人工智能·深度学习·神经网络