ai训练绘画入门:一台3060训出自己的画风,我的全过程和参数

ai训练绘画入门:一台3060训出自己的画风,我的全过程和参数
ai训练绘画插画:电脑屏幕上正在训练画风模型的数据曲线

别绕了:ai训练绘画没有想象中神秘,一块3060、三十张图、五小时,就能训出一个会你画风的小模型。关键是图要挑得狠、打标要敢删,参数照我下面这套抄就行。

我是白天上班、晚上画画的那种人,自学到能勉强看的水平花了两年。去年开始玩AI出图,最气的一件事是:不管怎么写提示词,图都不是「我的画风」。别人的模型用了半年,总觉得隔着一层。

今年三月我咬咬牙决定自己训一个。之前完全没碰过机器学习,代码水平停留在改别人脚本的程度,最后也跑通了。所以这篇不唬人,讲的都是一个普通人踩过的坑。装环境那步我纯按别人教程抄的,没资格讲,跳过。

训练到底在练什么,先想明白再动手

训练的本质是喂给模型一批带文字标注的图,让它学会「这组词对应这种画法」。图定上限,标定下限,参数只占三分。

我一开始的理解是「把图丢进去它就学会了」,错得离谱。真正花时间的是挑图和打标:三十张图我挑了一个周末,标准只有一条——这一张里有没有我想让它学会的东西。糊的、试水的、构图有硬伤的,删了就删了,凑数不如不给。打标用的是自动打标工具跑一遍,然后逐张手改,把它瞎猜的词全删掉,只留我想要的画风特征。这一步比调参数痛苦多了,但值得。

想通原理之后再看那些讲出图流程的教程就顺了,比如这篇新手四步出图流程,本质都是在控制「词和图的对应关系」。训练只是把这个对应关系焊死在模型里。

图的来源得干净:我只用了自己这两年画的原创,没掺任何临摹和网图。一来版权安心,二来临摹图里别人的笔触会把画风带偏。我试过混进五张临摹图,出来的味道立马杂了,删掉重训才恢复正常。

打标还有个小技巧:给这批图统一加一个自造的触发词,我编了个毫无意义的英文词缀在每张图的标注开头,出图时带上它,这批画风就被唤起来了。三十张挨个检查标注花了一小时,咖啡续了三杯。

还有个心态问题得提前说:训出来的小模型不会比你本人画得好,它只是你画风的平均值。平均值里有你的优点,也有你的毛病。我第一版模型学会了我画背景偷懒的毛病,气笑了。后来把三十张里画得最不用心的五张换掉重训,模型立刻精神了一点。喂什么学什么,跟带孩子一个道理。

我的实操参数和两次翻车

我的配置:30张图、学习率1e-4、dim16、每200步存一版,总共训1200步。3060显存12G刚好够,训练一轮约四十分钟。

硬件是块用了四年的3060,训练时显存吃到11.2G,风扇声像起飞,但不影响我同时挂个视频。第一次翻车出在学习率上,我心大手笔调了2e-4,训出来的图整个糊掉,颜色溢出,像隔了一层油。退回1e-4就正常了。第二次翻车是过拟合:我贪心训了2000步,结果不管什么提示词,出来的人都长同一张脸。最后定在1200步,每一版都留档,哪版好用拿哪版。

分辨率我统一裁到768见方再训,宽高比太乱会拉低效果。batch size给2,显存正好卡满。存档命名我一开始走得野,v1到v6全靠后缀区分,后来自己都分不清哪版是什么参数,专门建了个表格记下来。这个习惯建议第一天就养成,省得半夜翻文件夹考古。

验证模型我有个土办法:拿十条没参与训练的提示词去测,每条出四张,二十分钟看完。其中两条专门写没练过的题材,比如雨夜和厨房,看模型会不会强行把画风糊上去。有一版连厨房灶台都要给你排斜线,这版直接淘汰。判断标准不是「像不像我的画」,而是「我下次愿不愿意用它出图」。还有个更细的信号,我自己总结的:看我画风里标志性的斜线排线有没有出来。有,就成了;没有,接着调。

算下来整个流程五小时出头,其中四小时在挑图和打标。说白了这事不难,难的是舍得删自己的图。

现在我的图库里躺着一个只属于我的小模型,出图速度比我手画快一百倍,虽然它学走了我画背景偷懒的毛病,但也把我的配色那点小心思留下来了。两年手绘加五小时训练,这笔账怎么算都值。等哪天它把斜线排线也学全了,我就把手绘板收起来——开玩笑的,手还是得画。

常见问题

显卡不行能训练吗?

3060 12G是舒服线,更低可以用云平台按小时租,几块钱能跑完一轮。纯CPU就别想了,等的功夫够你手画十张。

图越多效果越好吗?

不是。三十张挑得狠的,胜过三百张随手截的。图多了反而互相打架,模型学成四不像,先做减法再做加法。

训出来的模型能发出去或卖吗?

用的是你自己的原创图,分发一般没问题;里面混了别人作品或受版权保护的角色,就别碰商用,老老实实自己玩。

延伸阅读