fallcodde 发表于 2021-1-11 17:33:44

人脸自动筛选工具v0.3(for DFL2)
更多福利访问deepfacelab中文交流论坛dfldata.xyz

人脸自动筛选工具並非简单去重工具,即使表情角度不一样,它也可以在一堆头像中,筛选出与目标头像相同的人物。
本工具绝大部份代码来自坛主滚石的另一个帖子,这里主要是做了小部分修改和整合打包工作。感谢坛主的付出与努力。

白菜价2仙丹,希望大家喜欢
v0.2
更新算法,增加适用性
增加一个目录分类,无法分辨的人物会移动到unknown目录
v0.3
支持多样本作为检测参考,可以是不同的人,但是可能影响速度和准确率,参考样本不要太多(也不需要),建议不要超过5张
v0.4优化人脸识别方法,大大提升识别速度和可识别的角度。
可以放更多的参考样本,正脸侧脸大角度脸都放几张,建议5-10张,更多也行
美化界面+优化使用说明

安装说明:
1.解压缩之后会看见两个文件夹和三个文件
2.把它们全部复制到DFL的目录
3.首次使用需要先运行“安装_BBimageFilter.bat”,以安装运行所需环境。
4.安装时需要连接网络,安装脚本会自动下载一个100m左右的人脸特征库,请耐心等待完成。5. 去除安装包路径中的所有中文,此脚本不支持中文路径
6.正确安装后即可打开0) BBimageFilter.bat运行人脸自动筛选工具。
7.使用方法请参考工具内的说明
8.确定工具可以正常使用后,可以手动删除”安装_BBimageFilter.bat”与“dlib-19.19.0-cp36-cp36m-win_amd64.whl”这两个安装文件,不影响使用。

1.png



实际运行效果:

待处理的aligned文件夹如下,里面有迪丽热巴和白百何混合头像
QQ截图20210117204254.png
做为标准参考的白百何aligned头像放在如下std目录里
QQ截图20210117204316.png

运行脚本
QQ截图20210117210156.png

严格模式下被筛选出来的非白百何的图。 可以看到所有迪丽热巴图都被晒出来了,但也把部分白百何图也误杀了。
如果觉得过于严格,想保留更多大角度图,筛选时选择2中度或3宽松
QQ截图20210117210240.png


已有 1979 人购买  本主题需向作者支付 2 灵石 才能浏览 购买主题

滚石 发表于 2021-1-11 18:38:17

我那个代码写了好多硬编码,人脸框大小都写死了,估计适配性会有问题。我周末再帮你改一下

zdqssl 发表于 2023-8-2 15:51:29

Processing e:\deepfacelab\dfl_upto_rtx2080ti_98\dlib-19.19.0-cp36-cp36m-win_amd64.whl
dlib is already installed with the same version as the provided wheel. Use --force-reinstall to force an installation of the wheel.
WARNING: You are using pip version 21.3; however, version 21.3.1 is available.
You should consider upgrading via the 'E:\deepfacelab\DFL_UPTO_RTX2080Ti_98\_internal\python-3.6.8\python.exe -m pip install --upgrade pip' command.
Looking in indexes: https://pypi.tuna.tsinghua.edu.cn/simple/
Requirement already satisfied: face_recognition in e:\deepfacelab\dfl_upto_rtx2080ti_98\_internal\python-3.6.8\lib\site-packages (1.3.0)
Requirement already satisfied: face-recognition-models>=0.3.0 in e:\deepfacelab\dfl_upto_rtx2080ti_98\_internal\python-3.6.8\lib\site-packages (from face_recognition) (0.3.0)
Requirement already satisfied: Click>=6.0 in e:\deepfacelab\dfl_upto_rtx2080ti_98\_internal\python-3.6.8\lib\site-packages (from face_recognition) (8.0.4)
Requirement already satisfied: numpy in e:\deepfacelab\dfl_upto_rtx2080ti_98\_internal\python-3.6.8\lib\site-packages (from face_recognition) (1.19.3)
Requirement already satisfied: dlib>=19.7 in e:\deepfacelab\dfl_upto_rtx2080ti_98\_internal\python-3.6.8\lib\site-packages (from face_recognition) (19.19.0)
Requirement already satisfied: Pillow in e:\deepfacelab\dfl_upto_rtx2080ti_98\_internal\python-3.6.8\lib\site-packages (from face_recognition) (8.4.0)
Requirement already satisfied: importlib-metadata in e:\deepfacelab\dfl_upto_rtx2080ti_98\_internal\python-3.6.8\lib\site-packages (from Click>=6.0->face_recognition) (4.8.1)
Requirement already satisfied: colorama in e:\deepfacelab\dfl_upto_rtx2080ti_98\_internal\python-3.6.8\lib\site-packages (from Click>=6.0->face_recognition) (0.4.4)
Requirement already satisfied: typing-extensions>=3.6.4 in e:\deepfacelab\dfl_upto_rtx2080ti_98\_internal\python-3.6.8\lib\site-packages (from importlib-metadata->Click>=6.0->face_recognition) (3.10.0.2)
Requirement already satisfied: zipp>=0.5 in e:\deepfacelab\dfl_upto_rtx2080ti_98\_internal\python-3.6.8\lib\site-packages (from importlib-metadata->Click>=6.0->face_recognition) (3.6.0)
WARNING: You are using pip version 21.3; however, version 21.3.1 is available.
You should consider upgrading via the 'E:\deepfacelab\DFL_UPTO_RTX2080Ti_98\_internal\python-3.6.8\python.exe -m pip install --upgrade pip' command.

1427523462 发表于 2023-6-20 20:27:17

还是没那么好用,10万个图片,有4万个被挑出来,仰头侧脸错的最多,房间暗沉一点基本上全错,赵丽颖和赵露思是不是太像了错了很多,

x56987 发表于 2023-3-19 19:37:29

本帖最后由 x56987 于 2023-3-19 19:46 编辑

请问,出现下面的问题,启动时筛选不了,我的aligned和std都没问题,好像是少了face_recognition,这个是什么问题?需要安装么?怎么安装这个东西?



F:\AI\novelai-webui-aki-v2\models\Stable-diffusion

randal 发表于 2021-4-7 09:14:54

路过支持一下

just4u666 发表于 2021-1-16 17:57:37

求求大神出个ubuntu版本的:$

fallcodde 发表于 2021-1-12 11:30:17

crazyfox 发表于 2021-1-12 11:00
这个人脸大小是限制多少呢

人脸大小没有限制,512/1024都可以,欢迎反馈结果

chris2001 发表于 2021-1-11 19:25:02

本帖最后由 chris2001 于 2021-1-11 19:30 编辑

雪中送炭啊啊啊啊啊啊啊啊啊啊啊啊

我萌新,下版主的代码,光配置环境就花了整整一个白天。然后还是倒在一个imread的代码上面。网上搜索了一下,说要降级或者改新的代码。自己又不会改。刚想在求助区悬赏仙丹请高手改改。

打瞌睡正好遇上楼主这个大枕头。狠狠地给楼主个赞

雨夜屠夫 发表于 2021-1-11 17:36:33

沙发!!!!

zealer 发表于 2021-1-11 17:37:06

感谢分享!占个1楼!

jiaoxiangyu 发表于 2021-1-11 17:51:48

感谢分享....

hainaby 发表于 2021-1-11 18:19:32

这个必须支持

q888666233 发表于 2021-1-11 18:20:46

NB绝了学习了

fytxh26 发表于 2021-1-11 18:34:10

厉害。谢谢分享

poiu88 发表于 2021-1-11 18:40:22

必须支持必须支持
页: [1] 2 3 4 5 6 7 8 9 10
查看完整版本: 【人脸筛选】指定人脸自动筛选工具v0.4