利用数据增强让DF模型有比LIAE更好的色彩 by 菜级玩家

caiji132156

星级打分

• 1
• 2
• 3
• 4
• 5

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

将增强过的图片作为输入，让模型强制重建原图，把异常颜色作为干扰点让模型学会“从异常颜色的人脸恢复出正常颜色的人脸”

这样就可以避免解码器不会生成各种各样的颜色的人脸而导致的解码困难
此外，我还加了更激烈的数据增强，重建目标依然是原图（loss 2.72是因为我开着VGG）

可以看见，模型在极度夸张的情况下仍然可以重建 正常颜色的人脸，然后只需要用简单的调色手段，就可以实现更好的色彩

下面是效果：



这样模型有DF的高相似度，同时有比LIAE更好的色彩，并且没有LIAE那么臃肿、难训练

相关代码请访问我的项目：https://github.com/CaijiPlayer258/DeepFaceLab-PyTorch

caiji132156

训练直出，换脸效果是什么就是什么