Pytorch移植与更新

freezhh · 发表于 2022-7-1 17:28:19

星级打分

1
2
3
4
5

平均分:NAN 参与人数:0 我的评分:未评

我把leras移植成pytorch框架了，切脸换成了insightface，快了很多

天天海底捞 · 发表于 2022-7-1 18:42:31

可以详细介绍一下不

dsyrock · 发表于 2022-7-1 19:42:10

大佬！效果怎么样，有计划公开吗？

freezhh · 发表于 2022-7-1 20:25:11

主要是目前好多新的算法都是基于pytorch的，基于tensorflow的leras很难和其他算法通用，就移植了一下，并把原来切脸的算法换成了insightface包，并把论坛的一些神丹迁移了到了pytorch上，包括滚石的遮罩

freezhh · 发表于 2022-7-1 20:30:27

for face in faces:

      embedding = np.array(face.embedding).reshape((1, -1))
      embedding = preprocessing.normalize(embedding)

      if self.do_reg:
         r = self.feature_compare(embedding, self.feature)

         if not r:
            continue

      # reg first

      rect = np.array(face.bbox).astype(np.int32)

      detected_faces.append(rect)

      lmk = face.landmark_2d_106

用insightface切脸的好处可以把人脸识别加入到切脸中，自动识别要换的脸

zuishuai · 发表于 2022-7-1 20:55:58

提示: 作者被禁止或删除内容自动屏蔽

freezhh · 发表于 2022-7-1 21:22:24

本帖最后由 freezhh 于 2022-7-1 21:42 编辑

训练部分改了df， amp还没有

https://www.dfldata.xyz/forum.ph ... tid=7426&highlight=范冰冰

对这个模型进行了训练，merge，都做了

freezhh · 发表于 2022-7-1 21:24:55

df训练加入了insightface中的id loss

港城黎明 · 发表于 2022-7-1 22:18:30

大佬您QQ多少啊您在交流群吗？

这是个好家伙 · 发表于 2022-7-1 22:33:55

torch用起来确实比tensorflow好些，不过你这门槛太高了，不专业的都没法弄。

		自动登录	找回密码
密码			立即注册（仅限QQ邮箱）