ClothingMigration-NCNN
What
- CT-Net:CVPR2021上的一个虚拟试穿/服装迁移的工作,该工程就是将CT-Net及其相关依赖网络移植到ncnn
- 服装迁移:给定一个人的图A,与另一个人的图B,将A的衣服穿到B的身上
PS
这个工作其实我个人计划做的一个系列,分三步:
Introduce
这个没啥好说,看图就完事了,先看换男装的: 再看女装: 程序流程分析,假定目标服装图A,目标穿衣人B:
- 对A与B都应用OpenPose得到candidate和subset
- 对A与B都应用JIP_LPPNet得到segment
- 对B应用DensePose得到IUV
- 对两个candidate、两个subset、两个segment、一个IUV以及图A、B应用CT-Net即可得到换装后的结果
Resource
- 上图的集成了所有功能的exe程序,可以从这里,提取码:6666获取,由于一共集成了4个模型,因此非常大,有1个多G,同时运行也很缓慢,大约要一分钟
- 该github工程仅包含代码,实际运行调试需要模型,可以从这里,提取码:6666获取,再放到对应的位置就好了
- 知乎分析文章:https://zhuanlan.zhihu.com/p/452366369
Content
将CVPR2021的服装迁移模型CT-Net搬运至NCNN框架上
CT-Net依赖
- 依赖OpenPose的人体姿势——pytorch实现
- 依赖LIP_JPPNet的的人体解析分割图——Tensorflow实现
- 依赖Densepose的人体IUV图——Detectron2实现
待完成&已完成
- OpenPose的ncnn&c++实现 —— OpenPose文件夹
- LIP_JPPNet的ncnn&c++实现 —— LIP_JPPNet文件夹
- Densepose的ncnn&c++实现 —— DensePose文件夹
- CT-Net的ncnn&c++实现 —— CT-Net文件夹
- 组合四个模型的ncnn&c++实现 —— Ensemble文件夹
- QT GUI实现 —— QT文件夹