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

89542767 發(fā)布于2022-11-12 22:24 / 426人閱讀

　　Python的用處還是比較的大的，在工作當(dāng)中，方方面面的都會(huì)遇到使用Python這門技能。那么，怎么實(shí)現(xiàn)圖形之間的轉(zhuǎn)換呢？比如，將我們平常的頭像，轉(zhuǎn)換成為動(dòng)漫風(fēng)格的呢？下面就給大家詳細(xì)解答下。

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

　　PyQt5框架

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

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

　　Name:可以自己自定義

　　program:QtDesigner的安裝路徑，小編這里引用的是Anaconda下面安裝的PyQt5-tools的相關(guān)文件

　　parameter:不填

　　directory:

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

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

　　其中常用的控件有：

　　Push Button:按鈕

　　Line Edit:文本編輯框

　　Label:標(biāo)簽

　　Horizontal Line:水平線

　　Graphics View:繪畫

　　代碼邏輯

　　當(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)擊“開始”按鈕的時(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("開始加載圖片.......")
　　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來跑這個(gè)模型，所以整體運(yùn)行下來需要點(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)

　　然后我們開始運(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)程