【预训练】224-256-AMP 50W迭代可用于直播换脸

night163 · 发表于 2021-11-14 22:35:37

星级打分

1
2
3
4
5

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

本帖最后由 night163 于 2022-9-5 23:59 编辑

2020/9/5 更新
更新了失效链接

上次我传了一个参数相对高一点的预训练模型，可能大家没有那么强的显卡，效果虽然还行，但是跑起来有点吃显卡，而且直播换脸帧率也上不去。
这次又炼了一个默认参数的预训练模型，测试了一下，在6G显存的1660s上、12g显存的3060上，都可以跑起来。
以下为模型参数（其实就是默认的amp参数）：
================== Model Summary ===================
==                                              ==
==          Model name: defult_AMP             ==
==                                              ==
==    Current iteration: 503788                ==
==                                              ==
==---------------- Model Options -----------------==
==                                              ==
==          resolution: 224                   ==
==          face_type: wf                   ==
==    models_opt_on_gpu: True                   ==
==             ae_dims: 256                   ==
==          inter_dims: 1024                   ==
==             e_dims: 64                   ==
==             d_dims: 64                   ==
==          d_mask_dims: 22                   ==
==       morph_factor: 0.5                   ==
==          uniform_yaw: False                ==
==       blur_out_mask: False                ==
==          lr_dropout: n                      ==
==          random_warp: True                   ==
==             ct_mode: none                   ==
==             clipgrad: False                ==
==    autobackup_hour: 0                      ==
== write_preview_history: False                ==
==          target_iter: 1000000000000          ==
==    random_src_flip: False                ==
==    random_dst_flip: True                   ==
==          batch_size: 8                      ==
==          gan_power: 0.0                   ==
==       gan_patch_size: 28                   ==
==             gan_dims: 16                   ==
==                                              ==
==------------------ Running On ------------------==
==                                              ==
==       Device index: 0                      ==
==                Name: NVIDIA GeForce RTX 3090 ==
==                VRAM: 18.35GB                ==
==                                              ==
====================================================

试了一下直播换脸帧率，1660s还是可以保证20帧的，完全足够日常使用。
复用任何脸型都非常快。
如果要使用，删掉inter_dst文件，然后正常训练。
如果你觉得你训练出来的脸已经很像了，可以关掉random_warp，然后打开 lr_dropout再训练一段时间，效果更好了以后打开gan，根据你的显存来。
我自己感觉色彩到最后再开比较好，我觉得lct的效果相对更好一点。

用于检测是否失效，如果失效请及时联系我