pytorch开发的仿DFL的换脸软件

day270010678 · 发表于昨天 18:48

星级打分

1
2
3
4
5

平均分:NAN 参与人数:0 我的评分:未评

本帖最后由 day270010678 于 2026-9-14 20:26 编辑

这是源码，如果有人无聊，没事做的话可以封装为exe，提供给大家，我这里这提供bat启动run_app_swap.bat。
把DFL的功能基本都完善了，支持DFL的权重文件的转换，如果用dfl的权重，记得在工具面板里面转换下。
另外我复现了人脸提取权重的训练，所以你可以训练自己的专属权重文件，workspace\insightface_train\pretrain_faces\no_face，这个里面就是训练的时候放的负样本数据，记住这个目录不能放任何有人脸的图片，就是把没有人脸的图片放这里面，帮助权重去识别以前识别错误的图片。
另外workspace\insightface_train\pretrain_faces\occlusion_masks目录就是放遮挡物的图片，一定是遮挡物，不是放人脸，比方说头发，水杯，手，还是xx等等遮挡面部的东西，带 alpha 的 png，**说明：** 生成时统一 256×256、带 alpha 通道。训练时会被缩放（脸宽的 0.3~0.6 倍，即遮挡块约 58~115px @192 输入），所以这个分辨率足够。。**换真实素材时的建议：** 分辨率 ≥256 更好（≥512 更佳，缩放后边缘细节保留多）、主体居中、边缘带羽化（alpha 渐变），这样贴到脸上边界自然、不被看出 "贴图感"。因为本人ps能力有限，不会扣图，不会把手等遮挡物抠出来，所以只生成了几个简单的图片。这些都是权重训练的数据集。
软件的使用说明，参数放docs目录了。
另外着重说明：因为我弄了很多功能，这个DFL版的纯粹是为了仿造DFL，在转为安装包的时很多功能都暂时剔除了，加上我以前的源码有点乱，所以我很小心的把以前的代码都尽可能的优化加整合了，理论上应该不会出现bug，但是世界上没有绝对的事情，如果出错了，你们自己修改下，或者你说下，我修改。
另外着重说明一点，训练，我没有完整的训练，我只训练了几个小时，十几万的迭代，没看出问题，所以就完整的封装了，理论上不会出错，但是不保证100%，如果出错，记得提醒下。
至于功能，看大家需要什么，或者希望增加什么功能，或者希望更好的体验方面。可以留言，空了就修改，或者大家修改后，告诉我。喜欢的就可以用。
关于数据集，我复现insightface的训练的是，仅仅是用了lapa和wflw两个数据集测试的，我也上传了，如果需要的可以自己下载去训练，解压放到workspace\insightface_train\pretrain_faces里面就行了。
另外关于to_annotate目录，这个是你从不同途径复制的希望以后提取用于insightface训练的图片，放到这个目录就行了，当然我这个里面也支持src和dst来源的。总之这三个来源的图片标注后会自动保存到manual_annotated目录，这个目录跟外部数据集目录pretrain_faces共同构成了insightface的训练的基石。
特别声明，不支持多gpu，不支持xpu，不支持amd，毕竟我也没有这些卡。多gpu支持同理，没有多gpu，没法测试，就不实现了。
另外根据torch 编译的 sm 列表 + ORT 的 CUDA 版本：
arch list: ['sm_75', 'sm_80', 'sm_86', 'sm_90', 'sm_100', 'sm_120']所以理论上支持N卡全系列。

百度网盘通过网盘分享的文件：DeepFaceLab_PyTorch
链接: https://pan.baidu.com/s/1YFJqEzvCwbIg6X4amZu8tA?pwd=xmzs 提取码: xmzs

阿里网盘[color=rgba(0, 0, 0, 0.85)]DeepFaceLab_PyTorchhttps://www.alipan.com/s/PZJa6jibkJe点击链接保存，或者复制本段内容，打开「阿里云盘」APP ，无需下载极速在线查看，视频原画倍速播放。。

郑重声明：仅仅是用于AI学习交流，禁止用本软件进行任何违法的行为。