python CodeFormer 图像（人脸面部）修复源码

介绍
github地址：https://github.com/sczhou/CodeFormer

[NeurIPS 2022] Towards Robust Blind Face Restoration with Codebook Lookup Transformer
效果：

?

测试环境：

anconda3+python3.8

torch==1.9.0+cu111

pyqt5

部分代码：

import math
import threading
import time

import cv2
import os
from PyQt5 import QtCore, QtGui, QtWidgets
from PyQt5.QtGui import QImage, QPixmap
from PyQt5.QtWidgets import QLabel, QFileDialog
from CodeformerManager import *


class MainForm(QtWidgets.QMainWindow):

    def setupUi(self):
        self.setObjectName("Form")
        self.resize(1320, 703)
        self.cb_background_enhance = QtWidgets.QCheckBox(self)
        self.cb_background_enhance.setGeometry(QtCore.QRect(20, 30, 91, 19))
        self.cb_background_enhance.setObjectName("cb_background_enhance")
        self.cb_background_enhance.clicked.connect(self.cb_background_enhance_clicked)
        self.cb_face_upsampling = QtWidgets.QCheckBox(self)
        self.cb_face_upsampling.setGeometry(QtCore.QRect(140, 30, 111, 19))
        self.cb_face_upsampling.setObjectName("cb_face_upsampling")
        self.cb_face_upsampling.clicked.connect(self.cb_face_upsampling_cliecked)
        self.label = QtWidgets.QLabel(self)
        self.label.setGeometry(QtCore.QRect(260, 32, 72, 15))
        self.label.setObjectName("label")
        self.sb_rescale = QtWidgets.QSpinBox(self)
        self.sb_rescale.setGeometry(QtCore.QRect(350, 28, 46, 22))
        self.sb_rescale.setObjectName("sb_rescale")
        self.sb_rescale.setMinimum(2)
        self.sb_rescale.setMaximum(4)
        self.sb_rescale.setValue(2)
        self.sb_rescale.valueChanged.connect(self.sb_rescale_value_changed)
        self.hs_fidelity = QtWidgets.QSlider(self)
        self.hs_fidelity.setGeometry(QtCore.QRect(600, 30, 311, 22))
        self.hs_fidelity.setMaximum(100)
        self.hs_fidelity.setProperty("value", 50)
        self.hs_fidelity.setOrientation(QtCore.Qt.Horizontal)
        self.hs_fidelity.setObjectName("hs_fidelity")
        self.hs_fidelity.valueChanged.connect(self.hs_fidelity_changed)
        self.label_2 = QtWidgets.QLabel(self)
        self.label_2.setGeometry(QtCore.QRect(420, 30, 171, 16))
        self.label_2.setObjectName("label_2")
        self.label_3 = QtWidgets.QLabel(self)
        self.label_3.setGeometry(QtCore.QRect(920, 30, 41, 20))
        self.label_3.setObjectName("label_3")

        self.pb_import = QtWidgets.QPushButton(self)
        self.pb_import.setText("导入图片")
        self.pb_import.setGeometry(QtCore.QRect(960, 28, 80, 30))
        self.pb_import.setObjectName("pb_import")
        self.pb_import.clicked.connect(self.import_image)

        self.pb_start = QtWidgets.QPushButton(self)
        self.pb_start.setText("开始修复")
        self.pb_start.setGeometry(QtCore.QRect(1050, 28, 80, 30))
        self.pb_start.setObjectName("pb_start")
        self.pb_start.clicked.connect(self.start_restore)

        self.label_src = QtWidgets.QLabel(self)
        self.label_src.setGeometry(QtCore.QRect(10, 80, 631, 571))
        self.label_src.setStyleSheet("background:rgb(0, 0, 0)")
        self.label_src.setText("")
        self.label_src.setScaledContents(True)
        self.label_src.setObjectName("label_src")
        self.label_dest = QtWidgets.QLabel(self)
        self.label_dest.setGeometry(QtCore.QRect(650, 80, 661, 571))
        self.label_dest.setStyleSheet("background:rgb(0, 0, 0)")
        self.label_dest.setText("")
        self.label_dest.setObjectName("label_dest")
        self.label_dest.setScaledContents(True)

        self.retranslateUi()
        QtCore.QMetaObject.connectSlotsByName(self)
        self.init_all()

?使用方法：

安装好python环境，然后执行代码

python main.py即可出现界面

下载地址：

https://download.csdn.net/download/FL1623863129/88665053