利用Python实现一键将头像转成动漫风

# 选择本地图片上传
def openImage(self):
    global imgNamepath  # 这里为了方便别的地方引用图片路径，将其设置为全局变量
    imgNamepath, imgType = QFileDialog.getOpenFileName(self.ui, "选择图片", "D:\\", "*.jpg;;*.png;;All Files(*)")
    img = QtGui.QPixmap(imgNamepath).scaled(self.ui.label_4.size(), aspectMode=Qt.KeepAspectRatio)
    print("img: ", img.width(), img.height())
    self.ui.label_4.setFixedSize(img.width(), img.height())
    # 在label控件上显示选择的图片
    self.ui.label_4.setPixmap(img)
    self.ui.label_4.repaint()
    # 显示所选图片的路径
    self.ui.lineEdit.setText(imgNamepath)

def saveImage(self):
    img = self.ui.label_5.pixmap().toImage()
    fpath, ftype = QFileDialog.getSaveFileName(self.ui, "保存", "D:\\", "*.jpg;;*.png;;All Files(*)")
    img.save(fpath)

def startAction(self):
    net = Generator()
    net.load_state_dict(torch.load("./weights/face_paint_512_v2.pt", map_location="cpu"))
    net.to("cpu").eval()
    image = self.load_image(imgNamepath)
    print("开始加载图片.......")
    with torch.no_grad():
        image = to_tensor(image).unsqueeze(0) * 2 - 1
        out = net(image.to("cpu"), False).cpu()
        out = out.squeeze(0).clip(-1, 1) * 0.5 + 0.5
        out = to_pil_image(out)
    image_name = imgNamepath.split(".")[0]
    out.save(image_name + "_animegan" + ".png")
    print("图片保存成功！！")
    imgShow = QtGui.QPixmap(image_name + "_animegan" + ".png").scaled(self.ui.label_5.size(), aspectMode=Qt.KeepAspectRatio)
    self.ui.label_5.setFixedSize(imgShow.width(), imgShow.height())
    self.ui.label_5.setScaledContents(True)
    self.ui.label_5.setPixmap(imgShow)
    print(f"image saved: {image_name}")

class Pic1:
    def __init__(self):
        self.ui = QUiLoader().load('pic_1.ui')
        self.ui.pushButton.clicked.connect(self.openImage)
        self.ui.pushButton_2.clicked.connect(self.startAction)
        self.ui.pushButton_3.clicked.connect(self.saveImage)

if __name__ == '__main__':
    app = QApplication([])
    # 显示创建的界面
    MainWindow = Pic1()  # 创建窗体对象
    MainWindow.ui.show()  # 显示窗体
    app.exit(app.exec_())  # 程序关闭时退出进程