人脸提取性能大提速,4X速度质量不变提取

deepfakelove

星级打分

• 1
• 2
• 3
• 4
• 5

平均分:5  参与人数:1  我的评分:未评

亲测,在3090上提速4倍
第一次玩换脸，我提取了一万多张用于训练的相片，想提取人脸，结果看到1s/it，又看到自己cpu gpu占用率都很低，于是就着手研究如何并行化提速
首先打开cmd,注意红色文字是我的文件路径,然后执行python路径
cd E:\ai\DFL_maozhihanhua_RTX3000\_internal\
e:
./setenv.bat  


这时候你就可以执行命令行来提取文件了
cd E:\ai\DFL_maozhihanhua_RTX3000\_internal\python-3.6.8\
python E:\ai\DFL_maozhihanhua_RTX3000\_internal\DeepFaceLab\main.py extract  --detector s3fd --input-dir E:\ai\DFL_maozhihanhua_RTX3000\workspace\data_src\ --output-dir E:\ai\DFL_maozhihanhua_RTX3000\workspace\data_src\aligned --no-output-debug --face-type whole_face --max-faces-from-image 0 --image-size 512 --jpeg-quality 90 --force-gpu-idxs 0

标红的地方都是我环境

现在既然你能够命令行执行提取文件夹了，也就自然可以把文件内容分几个文件夹，然后执行几个这个命令同时进行

当然，这样有点傻，每次自己要创建几个文件夹和输出文件夹，然后托文件分均匀，然后改几次命令，然后生成好在都拖出来

如果你还想更省事，这时候就需要我这个花费十五分钟写的自动化脚本，自动拷贝，自动均匀分布，自动多线程开命令，全部完成后自动合并，唯一需要的就是您来指定开几个进程.因为每个人显存不同，开多了显存会爆，而且这个应用好像没有给我简单设置每一个显存默认值的地方(至少我花费20分钟并没有找到)

下面就是我的脚本，希望您尊重他人劳动成果，有问题可以在下面留言，但我并不保证我能快速回复


关于如何自己动手更改线程数 看下图红箭头处的内容: 这是4个线程的时候的几个值，如果你要改成5个就所有的位置都+1 即可




python代码截图示例

z_charles

我来试试

hhgg2277

这个不错啊，切脸太费时间了，尤其是性能不行的。

百里瑾轩

有什么办法可以提升自动提脸的精度？好多脸提不出来需要手动提取太痛苦了。

king00

66666

Deannnn

我上次也找了下，还以为有隐藏参数啥的，果然还是得多线程。支持一下lz

fanda

刀哥有搞过这个，21年的左右，可以吃满显存，多线程适合显存大的显卡。

不过刀哥的是只支持 Directx_12版本

deepfakelove

fanda 发表于 2023-8-9 10:12
刀哥有搞过这个，21年的左右，可以吃满显存，多线程适合显存大的显卡。

不过刀哥的是只支持 Directx_12版 ...

我这个脚本应该是无关乎版本的，就调用命令行执行就行了

deepfakelove

Deannnn 发表于 2023-8-9 09:30
我上次也找了下，还以为有隐藏参数啥的，果然还是得多线程。支持一下lz

应该有一些调优，只要找到ts的参数设置下，就可以不用被限制在4倍速度上，这个显卡内存完全没必要启动吃8G的，不知道为啥在我24G的卡上上来就吃8G

jaysfing

请问这个相当于还是用的原版切脸对吧  精度是否和原版一致？谢谢