把模特练成一个文件:ai生成模特模型训练全记录
先给判断:训练专属模型,素材占八成,参数占两成,翻车十有八九翻在照片上。下面是我的技术宅自白:把店里用了半年的ai模特练成了一个两百多兆的模型文件。三关验证、三组参数、电费成本账,都在这篇里。
ai生成模特模型这个念头压了小半年才动手,结论先放这儿:月出货过百再练,练完单张图成本压到几毛;小体量的店用参数固定就够,别烧这个钱。
交代身份。大学学的计算机,毕业写了几年代码,现在开男装网店当副业,白天对账,晚上折腾显卡。店里那套ai模特图,之前靠设定卡和长期档案撑着,seed锁构图,档案管容貌。坏在上个月上秋装,我想让"她"换二十套衣服出片,档案法开始露怯:脸还认得,姿势神态越来越木,套套图片像复制粘贴。玩ComfyUI的老同学说了句"你不缺参数,缺一个自己的模型",我回家当晚就把硬盘翻了个底朝天。
这篇就是完整记录,数字全是真的,包括翻车那段。
训练前置:素材集就得按清单来
素材集是训练的地基:30张上下、角度铺开、分辨率1024起步,比参数重要得多。教材烂,学生再聪明也考不好。我第一版就栽在这,后面细说。
清单直接抄给你。总量30张,别贪多,20张精选好过50张乱喂,这话后面还会再提一次,因为代价不小。分配上:正面15张、四分之三侧8张、正侧4张、低头或抬头3张。表情拆开:平静12张、微笑10张、闭眼或其他8张。光线三档:柔光棚感15张、自然光10张、逆光或侧光5张。
几条硬规矩。背景和衣服每张都得不一样,同一件衣服超过3张,模型会把那件衣服也"背"进脸里——真事,我的第一版模特换什么上衣领口都长一个样。分辨率1024起步,短边低于768的扔掉,人占画面三分之二到五分之四,别切关节。
攒这30张花了四个多小时,比选品还费劲。来源只有一个:店里之前用档案法出的自有形象图,张张自己攒的,版权干净。这点我较真——拿别人的照片练模型,练出来的东西说不清归谁,商业上不能碰。
训练过程:学习率和步数,我试了三组
学习率8e-5、步数1800、余弦退火,是我三组试下来最稳的一档;1e-4太猛,5e-5太温,抄作业直接抄这组。取舍过程往下看。
底模用的开源SDXL系,LoRA秩32,batch size 2,文本编码器不动。参数没瞎猜,列了三组对照,一晚跑一组。
A组,学习率1e-4、步数2000:1500步后脸像了,皮肤质感开始塑料化,磨皮磨到亲妈不认识,十张废六张。B组,学习率5e-5、步数2500:稳是稳,像度死活上不去,十张里认得出脸的不到六张,典型欠拟合,白烧一晚电。C组,学习率8e-5、步数1800、余弦退火:像度和质感同时在线,定格,后面全用这套。
训练时长顺嘴交代:1024的图在4070上一步约5秒,1800步连验证带存档两个半小时,三组全跑完显卡烧了七个多小时。我的流程是睡前点开始,起床看图,白天开店晚上排任务。跑题一句,排训练有种开店的熟悉感,摁住变量、等结果、复盘,一个道理。咳,拉回来说验证。
验证:新模型得过三关
三关全过才准上岗:同脸关看像不像,换装关看穿新衣认不认,场景关看换地方崩不崩。我的成绩是九成、八成、七成。
我写了个小脚本抽10组提示词盲测,比对工具算人脸相似度,三关标准和结果如下,及格线八成:
| 关卡 | 过关标准 | 成绩 |
|---|---|---|
| 同脸关 | 10组不同提示词,相似度0.75以上算过 | 9/10 |
| 换装关 | 旗袍、西装、卫衣各出一组,脸不串妆不花 | 8/10 |
| 场景关 | 街拍、影棚、逆光三种光线,构图不崩 | 7/10 |
这张成绩单背后翻过一次车。第一版素材攒的时候图省事,30张里26张是同一个四十五度侧脸,出来的模型就只会这个姿势,让它转正面,五官直接换人,十张八张一个角度,跟连环画似的。行话叫过拟合,说白了教材只有一页,学生把那页背得滚瓜烂熟,换张卷子就傻眼。重攒素材、重跑C组,才有上面这张表。
补一句:场景关7/10是压线过的,崩的多是逆光特写,局部重绘能救,不算硬伤。
成本账:显卡时间、电费和我的晚上
直接报账:电费不到两块钱,显卡折旧忽略不计,大头是时间——素材加试参十二个小时上下,按外包时薪折算一千多。
细拆。显卡是自己的4070,满载三百瓦上下,七小时约两度电,五毛五一度算一块一,加整机功耗撑死两块。没卡的,云上按小时租同档卡,四块上下一小时,三组跑完三十块以内。
时间才是贵的那笔。攒素材四小时,写验证脚本两小时,三组参数三个晚上,加起来十二小时往上,按外包时薪折一千多。这是一版的价,第二版起素材流程都现成,之后每练一个新模特,就是一晚电费加一顿宵夜。
所以我的账本这么记:月出货过百、模特要换装换景的,这十二小时三周内摊平;一个月几十张图的,老实回去用seed加档案,把晚上留给打游戏。
写到这儿捋一遍:素材按清单攒,参数从抄作业开始,三关验完再上岗,成本算时间不算电费。档案法是日常,专属模型是节日,大促上新要批量出片的节点,我会把"她"拉出来加练一版。显卡还在机箱里嗡嗡转,下一版模特已经在排队了。
常见问题
不会写代码,能自己练吗?
能。现在有带界面的训练工具,点选文件夹就能跑,我这套里写代码的只有验证脚本,拿现成比对工具手工看图也能替代。花时间的仍是整理素材,这步外包不了。
素材最少要多少张?
我建议30张上下起步,往下探到15张也能练,但角度得铺得更开,翻车概率明显升高。质量永远压过数量。
用自己店铺的形象练模型,要注意什么?
素材必须全是自有或书面授权的形象,没有商量余地。练出来的模型别外传文件本身,商用范围以授权条款为准,我全部自己留着出店内图。
显卡带不动怎么办?
8G显存跑1024会比较挣扎,两条路:降到768凑合练,或者云上按小时租卡,四块钱上下一小时,练完销毁环境,比升级显卡便宜太多。