本帖最后由 comeon 于 2023-08-27 19:16 编辑
CodeFormer AI是一款非常好用的模型开源的视频去码、图片修复软件 处理效果展示
开源项目地址 sczhou/CodeFormer: [NeurIPS 2022] Towards Robust Blind Face Restoration with Codebook Lookup Transformer (github.com) 下面是本地部署教程(如果你的电脑有很好的gpu,没有的话cpu也可以,会烧) 1.安装环境: python (建议10左右的版本,太高报错) https://www.python.org/downloads/ 注意!安装时勾选add .exe to path!否则就得手动加进系统环境里 git https://git-scm.com/ conda https://www.anaconda.com/products/distribution Pytorch pytorch.org/ (根据自己的系统选择,选conda,python,显卡类型) ( 英伟达独立显卡,cuda引擎大于10.1。否则只能通过cpu解码) 2.安装其它需求包 cmd或conda powershell 逐条执行下列代码(在管理员权限下进行) git clone https://github.com/sczhou/CodeFormer cd CodeFormerpip install -r requirements.txt -qpip install -q gradiopython basicsr/setup.py develop 3.【下载预训练模型】(可能要加速): python scripts/download_pretrained_models.py facelibpython scripts/download_pretrained_models.py CodeFormer 这时我们就部署完成了! 4.推荐在conda powershell中使用各个功能(不容易出问题),结果将保存在results文件夹中。 面部修复(裁剪和对齐的面部) python inference_codeformer.py -w 0.5--has_aligned --input_path [input folder] 整体图像增强 python inference_codeformer.py -w 0.7 --input_path [image folder/image path] 视频增强 python inference_codeformer.py --bg_upsampler realesrgan --face_upsample -w 1.0 --input_path
|