meiyouruguo 发表于 5 天前

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

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 = 改成      resolutions =
第一次训练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--sourceexample/yangmi.jpg--target example/mabaoguo.jpg    --modelPathmodel/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 发表于 5 天前

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

mjy9921129 发表于 5 天前

和dfm模型有什么关系？

jandown 发表于 4 天前

感谢分享
这个能训练专模吗？
页: [1]
查看完整版本: Reswapper训练万能模型解压即用整合包