一直想把自己的脸做成能换城市换西装的那种图,折腾了下SDXL的DreamBooth训练。底模用的SDXL 1.0 Base,试下来做写实这个基座还是最稳。
最大的教训是数据集。我这批训练图质量只能算中等,背景老重复、衣服也重复、还缺全身姿势,结果就是过拟合、泛化差,生成时总跳出训练集里那几个背景。正则化图我挑的是高质量真人素材,这步偷懒不得。
还有个坑是epoch存点,我从150存的,出来的图明显有点训过头了。想省GPU的话Kaggle的免费notebook能跑,配置得算好步数再定保存频率。整体不难,就是细节多。