快捷導(dǎo)航

利用Python實(shí)現(xiàn)一鍵將頭像轉(zhuǎn)成動(dòng)漫風(fēng)

更新時(shí)間：2022年07月05日 15:03:44 作者：欣一

小編今天將為大家詳細(xì)介紹如何利用Python語(yǔ)言制作一個(gè)UI界面，大家可以通過(guò)一鍵點(diǎn)擊就實(shí)現(xiàn)頭像照片轉(zhuǎn)化成動(dòng)漫風(fēng)格的功能，感興趣的可以動(dòng)手嘗試一下

最近在Github上面有看到將頭像轉(zhuǎn)化成動(dòng)漫風(fēng)的項(xiàng)目，但是對(duì)于不少?zèng)]有技術(shù)背景的同學(xué)來(lái)說(shuō)可能就不知道該怎么使用了，小編今天制作了一個(gè)UI界面，大家可以通過(guò)一鍵點(diǎn)擊就實(shí)現(xiàn)頭像照片轉(zhuǎn)化成動(dòng)漫風(fēng)格的功能。

PyQt5框架

用Python編程語(yǔ)言來(lái)制作UI界面的框架有很多哈，大家原則上哪種框架用的順手就用哪種框架，小編這里使用的是PyQt的框架，因?yàn)樗С质謩?dòng)繪制控件，并且可以動(dòng)態(tài)加載，我們?cè)?code>Pycharm當(dāng)中添加上QtDesigner作為外部工具，來(lái)進(jìn)行界面設(shè)置，生成ui文件，步驟如下

其中需要填寫(xiě)的內(nèi)容當(dāng)中：

Name:可以自己自定義
program:QtDesigner的安裝路徑，小編這里引用的是Anaconda下面安裝的PyQt5-tools的相關(guān)文件
parameter:不填
directory:

在導(dǎo)入成功之后，我們打開(kāi)QtDesigner的初始界面

然后在界面的左邊，我們能看到一系列的空間，我們可以將這些控件拖拽到界面當(dāng)中進(jìn)行UI的設(shè)計(jì)，

其中常用的控件有：

Push Button: 按鈕
Line Edit: 文本編輯框
Label: 標(biāo)簽
Horizontal Line: 水平線
Graphics View: 繪畫(huà)

代碼邏輯

當(dāng)我們點(diǎn)擊“選擇圖片”實(shí)現(xiàn)圖片上傳的功能，代碼如下

# 選擇本地圖片上傳
def openImage(self):
    global imgNamepath  # 這里為了方便別的地方引用圖片路徑，將其設(shè)置為全局變量
    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)

同樣地，當(dāng)我們點(diǎn)擊“保存”按鈕的時(shí)候，實(shí)現(xiàn)圖片保存在本地的功能，代碼如下

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

而當(dāng)我們點(diǎn)擊“開(kāi)始”按鈕的時(shí)候便嘗試將上傳的圖片轉(zhuǎn)換成動(dòng)漫風(fēng)格，代碼如下

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("開(kāi)始加載圖片.......")
    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}")

代碼中我們是引用了預(yù)先訓(xùn)練完成的神經(jīng)網(wǎng)絡(luò)，并且默認(rèn)的是用電腦的CPU來(lái)跑這個(gè)模型，所以整體運(yùn)行下來(lái)需要點(diǎn)時(shí)間，這個(gè)根據(jù)不同電腦的硬件配置而定，其余的代碼如下所示

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)

然后我們開(kāi)始運(yùn)行整個(gè)界面

if __name__ == '__main__':
    app = QApplication([])
    # 顯示創(chuàng)建的界面
    MainWindow = Pic1()  # 創(chuàng)建窗體對(duì)象
    MainWindow.ui.show()  # 顯示窗體
    app.exit(app.exec_())  # 程序關(guān)閉時(shí)退出進(jìn)程

效果如下

最后的最后我們通過(guò)pyinstaller將其打包成exe可執(zhí)行文件，當(dāng)然代碼當(dāng)中我們用到了外部的第三方文件，因此打包的過(guò)程也會(huì)比單個(gè)文件的打包會(huì)更加的復(fù)雜一些，下一篇文章中小編再來(lái)詳細(xì)的說(shuō)明如何操作

以上就是利用Python實(shí)現(xiàn)一鍵將頭像轉(zhuǎn)成動(dòng)漫風(fēng)的詳細(xì)內(nèi)容，更多關(guān)于Python頭像轉(zhuǎn)動(dòng)漫風(fēng)的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章！

您可能感興趣的文章: