V2EX = way to explore
V2EX 是一个关于分享和探索的地方
现在注册
已注册用户请  登录
MozzieW
V2EX  ›  机器学习

求教如何训练 StyleGan 的 classifier(分类器)

  •  
  •   MozzieW · 2022-12-08 15:32:21 +08:00 · 713 次点击
    这是一个创建于 714 天前的主题,其中的信息可能已经有所发展或是发生改变。

    最近领了一个人脸编辑的任务,大概就是输入一张人脸然后调整为微笑、大眼睛,不一样的是我们要的是“痛苦”的表情,试了 StyleGan 相关的编辑,可以跑通官方提供的模型,但是 CeleA 没有“痛苦“属性,需要自己动手,训练以恶类似 celebahq-classifier-00-male.pkl 的模型。

    看了资料是说要训练一个分类器,我尝试用 ResNet50 跑了一个,但是无法导入。看了 StyleGan 论文才的分类器结构和 Discriminator 一样。大概理解可以用代码里的 D_basic 跑一个模型,然后使用?瞎折腾几天了,感觉方向对了,道路很曲折 /sad 。

    求大佬指点。刚接触机器学习,表示了解不多。

    目前尚无回复
    关于   ·   帮助文档   ·   博客   ·   API   ·   FAQ   ·   实用小工具   ·   3264 人在线   最高记录 6679   ·     Select Language
    创意工作者们的社区
    World is powered by solitude
    VERSION: 3.9.8.5 · 24ms · UTC 12:07 · PVG 20:07 · LAX 04:07 · JFK 07:07
    Developed with CodeLauncher
    ♥ Do have faith in what you're doing.