训练AI模特怎么训才不出废脸:数据集和参数踩坑实录

训练AI模特怎么训才不出废脸:数据集和参数踩坑实录
训练AI模特的数据集准备与训练参数实操过程

把话挑明:训练AI模特,废脸九成废在数据集,剩下的一成废在训练中途偷懒。素材干净、触发词独立、epoch按档存档,三件事做到位,一次通过率能从三成拉到七成。这篇不聊钱,只聊手上的活。

训练AI模特这件事我在代运营公司干了一年多,前前后后给五家店训过专属脸,瑜伽服、帆布包、冲锋衣的类目都碰过。结论先撂这儿:难点从来不是点按钮,是喂进去的照片和训练中途那几下判断。照片挑错,参数调得再花也白搭。

费用账市面上已经有人算得很细,这篇不重复。只讲手上的活:素材从哪来、多少张够、参数动哪几处、训完怎么验收。里面三个坑——人像虚化烙进模型、自动打标带出齐刘海、连拍图训出单角度脸——都是拿重训时间换来的教训。

数据集:素材来源排个序,张数有个甜点区

素材自己棚拍最稳,客户旧原片次之,网图一律不用;张数25到35张是甜点区,脸部要占画面四分之一以上。不是越多越好,超过60张,杂物和背景会一起被学进去。

三个来源我按优先级排过序,三家店的数据摆在一起看,差别非常直白:

来源成片质量要花的功夫风险
寄样衣自己棚拍高拍半天基本没有
客户提供旧原片中挑图半天连拍重复、角度单一
买家秀和网图低没法挑版权雷、光线杂

帆布包那家客户图省事,甩给我两百多张买家秀。我挑出40张喂进去,训完的模型出图一律带手机直出的灰调,提示词换了一轮也拽不干净。后来让客户寄样衣,找本地摄影棚拍了一下午,80张原片挑32张,一次过。这中间补拍花的钱,比两次训练费加起来还少。

张数这块我专门做过对比。同一个触发词、同一组参数,15张版和32张版各训一次:15张版正面脸是稳的,一喊侧身或者低头,脸就飘到另一个人身上;32张版抽检二十张,十八张能认出同一张脸。再往上我也试过,60张那批混了几张桌面摆拍,模型学会了在画面角落摆杂物。30张上下,够用。

裁切别偷懒。脸部占到画面四分之一以上再喂进去,全身照直接训的话,脸的信息量太稀,模型会转而去学衣服。衣服是出图阶段换的,训练阶段只需要人。

训练参数:我只动三个地方

触发词自造一个无意义组合词,学习率在默认档上砍半,epoch按档存档留到12个。其余参数保持默认,改得越多,越难定位问题出在哪一步。

触发词别用现成单词。我用的是一串念出来毫无意义的字母组合,就是为了不跟任何已有概念撞车。撞车的后果很直接:你在提示词里写这个词,模型联想到的是别的东西,脸就串了。

自动打标那个坑单独讲。训练平台默认会给每张图自动写标注,瑜伽服那家客户的模特留齐刘海,标注里每一张都带"齐刘海"三个字,训完的模型出图清一色齐刘海——客户卖的是发带,模特顶着刘海算怎么回事。后来我把自动标注关掉,手动只写触发词加极少量必要描述,再训就正常了。标注越啰嗦,模型越容易把偶然特征当成必然特征。

学习率我砍过一次。默认档训出来前几张看着挺好,跑过第十个epoch之后脸开始糊,五官往一种"标准脸"上收。砍半之后收敛慢一点,立体感保得住。epoch按档存档这个习惯最值钱:每跑完一个epoch存一版,训完拿固定提示词加固定种子,每版出一组九宫格。冲锋衣那次,第8版开始像人,第10版最自然,第12版笑容开始僵——最后交货交的是第10版。不存档的话,你只能拿到最后一版,好坏都认了。

训完先别交货:三道关过完再说

验收三道关:九宫格里八成是同一张脸、换二十条姿势提示词十六张能修、换背景色不出训练时的旧背景。任何一道不过,先回去查数据集,别急着调参数。

第一道,锁种子出九宫格,八张以上是同一张脸才算训成。第二道,换姿势,我固定用二十条姿势提示词扫一遍,十六张以上脸部不崩、只需要小修,就过。第三道最容易被忽略:换背景。训练集里背景单一的话,模型会偷偷把那个背景记成默认值,你在提示词里喊沙滩,它给你灰白影棚。数据集阶段留两到三种背景色,这一关基本白过。

验收不过的时候,先回头查数据集。我见过太多人卡在第二关就猛调参数,其实问题出在训练集里侧脸照片只有两张。参数是药,数据是饭,药治不了饿。

训好之后量产,提示词又是另一套写法,槽位怎么拆、词序怎么排,那篇拆得更细:模特AI填词怎么填?美工拆6个槽位,可用率三成到八成。模型训得再好,词填废了一样出废图。

回到开头那句:训练AI模特,功夫七成花在训练开始之前。素材来源、张数、裁切、标注,这四件小事做扎实了,训练本身反而没什么玄的。下次开训之前,先把数据集按上面几条过一遍,比多买两次训练次数划算。

常见问题

手机拍的素材能不能直接用?

能,但别用同一个人像模式拍一整批。人像模式的虚化会被模型当成风格学进去,出图背景永远是糊的。手机拍就关掉人像模式,光线打匀,成片跟相机的差距没你想的那么大。

训练集里可以放戴眼镜或戴帽子的照片吗?

跟你卖的东西冲突就不要放。冲锋衣那次的数据集里有几张戴墨镜的没剔干净,之后三成出图自带墨镜。饰品、帽子、墨镜类目尤其注意,模特脸上任何常驻物件都会变成鬼影。

一个训好的模型可以给几家店铺共用吗?

脸可以共用,数据集最好分开。不同类目的商品混在同一个数据集里,模型容易把A店的道具烙进B店的图。我们现在是脸通用、商品靠垫图出,分得清清楚楚。

训一次大概要多久?

托管平台跑下来40分钟上下,加存档多花几分钟。真正花时间的是前面挑图裁图,一个数据集我正常要弄大半天。功夫花在前面,训练那40分钟就是走个流程。

延伸阅读