Reswapper训练万能模型解压即用整合包

meiyouruguo

Reswapper训练万能模型解压即用整合包, 可训练万能模型让rope软件使用



项目地址:https://github.com/somanchiu/ReSwapper


电脑要求：
1、英伟达显卡8G以上
2、系统要求：win10、win11
容易出错的地方
软件路径不要包含中文
使用说明:
图片数据集放到哪里
在FFHQ文件夹中放你自己的图片, 可以用你自己的图片数据集, 几万张不同人脸不同角度的图片, 可单独训练女的模型或男的模型

如何训练模型
以256分辨率为例, 第一次训练, 预训练模型 "reswapper_256-1567500.pth" 在model 文件夹中, 点击  "训练.bat"  即可训练
下次训练的时候, 修改 train.py 文件第225行  modelPath = f"{outputModelFolder}/reswapper_256-1567500.pth"  这里的模型名称路径

如何训练512分辨率模型
在train.py 文件的第238行可以修改分辨率 ,如果想训练512分辨率, 可以 把   resolutions = [256]  改成      resolutions = [512]  
第一次训练512分辨率,因为512分辨率的没有预训练模型,需要从头开始训练     modelPath = f"{outputModelFolder}/reswapper_256-1567500.pth"    改成  modelPath = None
下次再训练512分辨率模型,  修改 modelPath = f"{outputModelFolder}/512分辨率模型名称.pth"


模型效果测试
点击 "人脸交换.bat"会执行下边代码, 自己训练的模型名称路径不要忘记修改,例如 reswapper_256-1567500.pth
python swap.py  --source  example/yangmi.jpg  --target example/mabaoguo.jpg    --modelPath  model/reswapper_256-1567500.pth  --outputPath   example/save.jpg  --resolution 256

如何转换成onnx格式模型让rope软件使用
先修改 pth2onnx.py 第三行的模型路径,  然后点击 "转换onnx格式模型.bat" , 名称修改为reswapper_256.onnx,
然后放到rope软件目录的model文件夹中,需要使用rope先锋3.1版本

如果是512分辨率的模型, pth2onnx.py 文件中 img_size=256 这个改成 img_size=512(目前rope先锋3.1版本只添加了256分辨率模型, 未来如果有好的512模型, rope软件会添加上)


下载地址: https://pan.baidu.com/s/1xVxZpMPA9us5_6-OhbHMFQ?pwd=fi49

AI001

大佬，有做视频教程学习吗？

mjy9921129

和dfm模型有什么关系？

jandown

感谢分享
这个能训练专模吗？