【霏艺所思】Deepfakes 代码笔记

https://github.com/deepfakes/faceswap
本来是打算学习GAN才看的。。。
后来改成看图像处理算法学习了。。。

当年想看看GAN网络,后来想想我用不到【除了制造假图片,这个能干嘛?】
然后对图像处理算法更多研究了。。。
3
分享 2020-05-17

3 个评论

霏艺Faye 图书管理员
想学习 Deepfakes with GAN的同学,看这个开源项目
https://github.com/shaoanlu/faceswap-GAN

说明文件
https://github.com/shaoanlu/faceswap-GAN/blob/master/README.md
写的非常好,非常有学习价值!

---------------------------------------
deepfakes的目录结构
入口函数在scripts目录下面{extract,train,convert} 对应文件里的process函数
学习AI算法的同学,自己看train.py
学习图像处理算法的同学,自己看convert.py

具体实现在plugins目录下面
神经网络在train目录
图像处理convert目录

神经网络部分
网络结构定义 代码在plugins\train\model目录下实现 例如 original.py
看encoder和decoder实现   有些可能不叫这个名字,就搜索add_networks函数,看看add_network加了什么网络进去,反正学习网络结构而已

网络训练部分:
只看一句话 compile_predictors 就可以了
其中 optimizer 指向了
lib\model\optimizers.py 的 Adam 最终调用的是 keras的Adam实现
想换SGD的同学,得自己改改,增加对应的实现代码 在 get_optimizer

其中 loss 指向了
plugins\train\model\_base.py 的 loss_dict 里
要么直接就是调用了keras的loss实现
要么就是lib\model\losses.py 里的封装
需要自己增加losses,或者修改的,自己在loss_dict里增加或者找到对应的函数去修改
重点看下DSSIMObjective??? 反正我无聊看看这个
搜索 loss_function  ,去修改对应的函数,比如现在默认是default="mae",


最后图像相关
plugins\convert目录下面
我觉得这里面的代码对我比较有价值。。。
毕竟,我就需要看看美颜,锐化,之类的功能罢了~~~
pcv1984 新注册用户
你好 本人是一个主要工作是Java程序猿 了解一点点机器学习 向你学习
霏艺Faye 图书管理员 回复 pcv1984 新注册用户
你好 本人是一个主要工作是Java程序猿 了解一点点机器学习 向你学习

沒事啊,不需要説自己是什麽工作

你喜歡説什麽就説什麽

不懂可以問

要发言请先登录注册

要发言请先登录注册

发起人

霏艺Faye 图书管理员

https://www.facebook.com/booklove.crown/

状态

  • 最新活动: 2020-05-17
  • 浏览: 3889