转随风dfl训练篇

woshini · 发表于 2024-9-15 16:53:43

本帖最后由 woshini 于 2024-9-15 17:09 编辑

从关了预训开始就算是正式的训练，而这个训练过程没有一个准确标准流程可以走，因为要视情况来调整参数达到最理想的效果；具体分为三个阶段，第一是学习成像阶段，第二调整阶段，第三收敛取真阶段；
第一阶段：学习成像
1、正式训练就会有两组数据，既SRC与DST，开训就把【 masked_training: True】【random_warp: True】开启，其它都关闭（如果SRC素材不全面要把【andom_flip: True】同时开着）
由于只开这两个，LOSS值会下降得很快，成像也会跟着出来，预览图只要关注第五列；三种情况：有些图换得比较准确，有些惨不忍睹，而有些会一片糊；换得准确的是因为SRC与DST对上图，这个基本没问题，惨不忍睹的是SRC缺与DST角度图，这个训练长时间点会改善但都不会太理想；一片糊的是由于模型缺角度导致，这个练时间长点会出来，但会非常非常慢，模型缺角度的问题应该在预训练时就要补全；
2、这样要练多久？看你觉得第五列图大部分图轮廓五观基本准确就行（不要等全部，每张图起跑线都不同，你要等到那些缺图缺角度的都准确那浪费的是自己的时间，除非你补全所缺的角度），也不要太在意清晰度，开着RW就不会太清晰
3、这时候把LRD打开，即【lr_dropout: True】【 masked_training: True】【random_warp: True】开启，其它都关闭，（之前开了【andom_flip: True】的要保留打开）
4、开了LRD后图会变清晰一些，训练一段时间，当看LOSS值下降得很慢或已经陆续有反弹迹象时为止；第一阶段可以结束
5、BS大小推荐为4（尽管你可以开16），经研究对比发现，在这一阶段BS4与BS8同样跑6小时，4的效果比8要好很多，而2效果不理想
第二阶段：调整
1、这时候如果有些眼睛对不上，或脸部侧脸轮廓不清晰的就可把【 masked_training: True】【 eyes_prio】【uniform_yaw】打开，把其它关闭，特别是【lr_dropout: True】【random_warp: True】【andom_flip: True】在这一阶段一定要关
2、有时遇到源脸比目标脸小，预览图上出的效果明显被拉胖的情况，你就要同时把【 true_face_power】打开，值不要太大，不然颜色会差别很大，推荐是0.001；这个可以从一开始就开着，主要看你源脸是否与目标脸轮廓差别很大；3、有些人会喜欢SRC的五观却要DST的轮廓的就可以把【face_style_power: 0.0001】打开调整，这样效果会更像目标轮廓，与【 true_face_power】刚好相反，所以不要一同开着
4、当源脸比目标脸小时，合成时会用到蒙版侵蚀羽化来做来边缘合成的衔接，这时候可以把【bg_style_power: 0.001】打开能学习些边缘背景溶合信息，值不要太大，不然会爆5、颜色调整也可以这时开着，但多数情况由于SRC素材脸颜色不一样的情况，就算怎么训练颜色都不会自然，一般我都是关闭，后期AE或PR调整；开RCT模式会出现有些脸色虽然相近了，但一些红唇会特别的深色，白黑差度非常大
6、上面的参数可以逐个单开调整，也可以全打开一并调整，只是需求与效率问题；
7、这一阶段要注意的是，所有参数只是调整，达不到很精细，比如【 eyes_prio】有些眼睛会大体上对位但不一定能完全正确，【uniform_yaw】轮廓虽然出来了，你会发觉并不是很像SRC人物，主要还是缺SRC的原因；
8、关于LOSS值，当你打开上面的任何一个参数LOSS都会有不同程度突然飘高，这个不用在意，每一项就像是一个层，LOSS值在每一层间，比如（只是比如）123层，LOSS层值是0.39~0.3、029.~2、0.19~0.1，所以不用在乎LOSS的高低，要看LOSS的下降变化，当LOSS值下降得非常慢或有反弹迹象一段时间时，基本就可以停了。当然你要求高或时间充裕可以一直练，软件在某一阶段学不下去时就会自动跳到下一层；在第一阶段也是同样，【 eyes_prio】【uniform_yaw】可以在第一阶段就打开（不建议，因为有些不需要调整的你开着就是浪费GPU的资源），一直跑时间够长LOSS值都会下去；这里要明白一个道理，假如SRC缺角度，就如拿着初中的课本去运用高中的知识你再怎么花时间都不会有理想的效果，这就是有些人就算把LOSS值降得很低而效果却不理想的原因
9、BS大小在这一阶段尽情的开满
第三阶段：收敛取真
1、当调整到自己认为理想时就可以到最后一步，这时你应该文件备份一下，因为GAN是很容易过头，也容易爆
2、把【 masked_training: True】【lr_dropout: True】【gan_power: 0.0001】打开，其它所有都关闭
3、GAN取值很关键，到底应该取多大也没有明确的说法，打个比如，10CM的尺度，每1CM有个刻度，最理想的点在某一个刻度上，GAN的值就像你往前走的步伐尺寸，假如理想点在6刻度上，你GAN开1能到达，开2能快点到达，开3能更快到达，开4速度更快，但却会错过，永远不能到达；而就算你开123虽然能到达但一不留意就会错过；所以取值范围最好以最小步伐让他跑（ GAN的值可以小到0.0001），时间虽然长间点，也让关注点有空间；
4、当你打开GAN时，这时LOSS应当在0.1以下；
5、那么要多久算结束？看第五列，头像会慢慢的有所变化，清晰度，阴影感会慢慢的更强，这时头像外围部分会出现彩色格子，会随时时间越来越明显，但只要不在头像之内就不影响，当你觉得清晰度满意了就结束，要是脸部出现马赛克那就表明已经过头了。要是出现彩格就算是爆了
6、BS大小，GAN很占GPU，逐个尝试显卡能承载的最大值

brianhiew · 发表于 2024-9-19 22:41:46

多谢分享~~

240sxluv · 发表于 2024-10-13 22:08:07

感谢分享！

yangjianqi · 发表于 2024-10-15 08:28:09

一直都不知道怎么操作，学习了

云行九天 · 发表于 2024-10-20 19:06:33

学习一下

		自动登录	找回密码
密码			立即注册