最近领了一个人脸编辑的任务,大概就是输入一张人脸然后调整为微笑、大眼睛,不一样的是我们要的是“痛苦”的表情,试了 StyleGan 相关的编辑,可以跑通官方提供的模型,但是 CeleA 没有“痛苦“属性,需要自己动手,训练以恶类似 celebahq-classifier-00-male.pkl 的模型。
看了资料是说要训练一个分类器,我尝试用 ResNet50 跑了一个,但是无法导入。看了 StyleGan 论文才的分类器结构和 Discriminator 一样。大概理解可以用代码里的 D_basic 跑一个模型,然后使用?瞎折腾几天了,感觉方向对了,道路很曲折 /sad 。
求大佬指点。刚接触机器学习,表示了解不多。