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

night163

星级打分

• 1
• 2
• 3
• 4
• 5

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

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的效果相对更好一点。


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

byl

来看看                          

DFL8938o9

看起来效果还行

bsw0121

看不出来效果

2lc63713e

感谢大佬分享

tktkl28

感谢楼主的分享

1111111

感谢分享

HXIAOH

发现一只九龄

sosadmin

我的3060 只有10帧

YifLee

感谢分享