opencv进阶 ——（九）图像处理之人脸修复祛马赛克算法CodeFormer

最新推荐文章于 2025-05-26 09:52:20 发布

fengqiao1999

最新推荐文章于 2025-05-26 09:52:20 发布

阅读量2.4k

点赞数 12

CC 4.0 BY-SA版权

分类专栏： opencv AI C++ 文章标签： opencv 图像处理人工智能

本文链接：https://blog.csdn.net/fengqiao1999/article/details/139027453

算法简介

CodeFormer是一种基于AI技术深度学习的人脸复原模型，由南洋理工大学和商汤科技联合研究中心联合开发，它能够接收模糊或马赛克图像作为输入，并生成更清晰的原始图像。算法源码地址：https://github.com/sczhou/CodeFormer

Face Restoration

Face Color Enhancement and Restoration

Face Inpainting

模型部署

如果想用C++进行模型推理部署，首先要把模型转换成onnx，转成onnx就可以使用onnxruntime c++库进行部署，或者使用OpenCV的DNN也可以。

1、可在以下地址下载模型：https://github.com/sczhou/CodeFormer/releases/tag/v0.1.0

2、下载CodeFormer源码，在工程目录下添加onnx转换python代码

import torch
from basicsr.utils.registry import ARCH_REGISTRY

if __name__ == '__main__':
    device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
    net = ARCH_REGISTRY.get('CodeFormer')