预训练模型的概念与用法

admin

什么是模型？
神经网络模型顾名思义就像人的大脑。这么讲虽然有点理想化了，但有助于大家理解：
新建一个模型就像诞生一个小孩，此时它虽然有脑子，但里面东西是全新的。
训练模型，就像教这小孩画画
训练的方式也和人很像，给他看图，让他照着画，学的好给奖励，学的不好给惩罚。
训练越多，这脑子学的就越多，画出来的人脸就越棒

什么是预训练模型？
预训练模型是“预选训练过的模型”的简称。
同样以小孩做比喻，现在你要画一个迪丽热巴画像，你有两种选择
1. 自己生个小孩，从零开始教他画画，直到他能画出迪丽热巴
2. 领养一个别人家已经学会画画小孩，他虽然没画过迪丽热巴，但他画过刘亦菲

你觉得哪个小孩能更快更好的画出迪丽热巴？当然是第2个

预训练模型也一样，别人训练过的模型已经掌握了大量人脸绘制技巧，你拿来稍微用自己的图片训练下，很快就能生成不错的图像。


那如何使用预训练模型？
只需三步第一步：下载一个预训练模型，放到model文件夹内
第二步：训练阶段选择该模型


第一步：下载一个预训练模型，放到model文件夹内
模型的获取路子很多，你可以在论坛模型分享区下载别人分享的模型
市场上也有人卖模型，你可以买付费的也可以拿免费的玩。但记住一点：模型的质量和价格没有必然联系，有人拿垃圾模型卖高价，也有人拿牛逼模型免费送。
萌新如果不知道别人模型质量怎么样
这里我推荐几家质量不错的：
1. 模型区，质量有高有低，我发的模型都是我自己用过的，质量好坏不说，起码免费或超低价，试试又不会怀孕

第二步：训练阶段选择该模型
开始训练，在选择要训练的模型（如果文件夹中还有更多）和要训练的设备（GPU / CPU）之后，等待片刻，会提示2秒钟内按enter回车以修改模型设置，按它进入参数设置环节
把最后一个选项（是否启用预训练模式 use pretrain mode）改为N。
有些模型已经关了该选项，那你就放心大胆用自己数据训练吧
有些模型没关该选项，关闭预训练模式后，模型会将迭代计数还原为0，这是预训练模型的正常行为，莫慌，主要看图是不是还有人影。
正常情况下，关闭预训练模式后，模型训练预览图是直接能看到八九不离十的人脸，如果你发现你的预览图又从一片完全看不出人形的灰褐色开始，莫慌，按以下步骤操作：
   1. 按enter回车结束训练
   2. 把下载的预训练模型原始文件中几个.npy文件（只要.npy格式的文件，其余文件不要动）复制替换到模型文件夹中
   3. 重新启动训练，把最后一个选项（是否启用预训练模式 use pretrain mode）改为N


然后就可以愉快的训练自己的人脸数据了


如何自己制作一个预训练模型呢?
有些老哥想自己调一个不寻常的模型结构，比如增加模型宽度，修改分辨率，那怎么才能自己制作一个预训练模型呢?
方法有两种
1. （推荐）正常创建模型，预训练模式选择n，然后src和dst的aligned文件夹中，放上你已有的人脸数据图片，src和dst重复也没关系，图越多越好，越杂越好。然后就挂机跑
2. （不推荐）创建模型时，预训练模式选择y。此时软件会使用一份自带的国外名人数据集来给模型训练。角度光影很丰富。但我不推荐这么做。一是这个数据主要是欧美人，缺亚洲人数据；二是自己想添加一些特定数据就很麻烦，还得去处理.pak打包文件；三是当你想替换为目标dst和src时，很容易没备份就直接改预训练模式为n，然后惊奇发现模型怎么变成回到初始状态了。几周的挂机就全浪费了，群里已有好几起这类惨案。

你自己如果缺预训练用的数据，可以去素材分享区找一个叫FFHQ的数据集，里面有很多各种各样光影条件下的人脸，而且清晰度特别高，适合用来做预训练。