CodeFormer的二次封装使用-针对人脸处理(需要有显卡最低4G显存)

CodeFormer能做什么

CodeFormer是一种基于AI技术深度学习的人脸复原模型，由南洋理工大学和商汤科技联合研究中心联合开发。该模型通过结合了VQGAN和Transformer等技术，可以通过提供模糊或马赛克图像来生成清晰的原始图像。它的主要功能有：

• 旧照片增强；
• 面部修复（图片不清晰）；
• 面部色彩增强和恢复；
• 脸部修复（照片缺角或残缺）。

查看显卡信息

nvidia-smi 

下载到本地解压缩后直接打开即可。

测试图片：

修复效果：

下载地址：https://download.csdn.net/download/feng8403000/88885979 

功能1、针对deepfacelab提取好的人脸图进行清晰化，直接保留deepfacelab相关切脸信息，强化后无需重新提取人脸，可直接用于训练。

功能2、对任意图进行清晰化处理，人脸部分采用codeformer算法处理，非人脸部分使用basicSR处理。适用于deepfacelab合成环节merge文件夹里的图，不适用aligned文件夹里的图。

功能3、直接对视频进行清晰化处理。默认只清晰化脸部，如需额外清晰背景，请勾选背景强化。

视频处理时间较久，请耐心等待。