2023年一月份跟二月份創(chuàng)建了一個PyQT5人工智能軟件開發(fā)系列的文章系列，過去的兩個月都沒怎么更新，心里一直想有時間繼續(xù)更新下去，今天又更新了一篇，基于PyQT5實(shí)現(xiàn)多線程、界面化、風(fēng)格遷移模型的實(shí)時推理。

基本設(shè)計思路

這個系列我好久沒有更新了，今天更新一篇PyQT5中如何實(shí)現(xiàn)風(fēng)格遷移模型的推理界面化與多線程支持。

參數(shù)設(shè)置選擇模型文件與數(shù)據(jù)文件地址作為輸入。 推理按鈕開發(fā)推理演示，支持視頻與圖像文件，開始推理會單獨(dú)開啟一個推理線程實(shí)現(xiàn)推理，推理結(jié)果通過信號發(fā)送到指定的槽函數(shù)處理之后更新界面，通過信號與槽機(jī)制實(shí)現(xiàn)界面線程跟推理線程相互獨(dú)立與數(shù)據(jù)共享。

模型說明

基于PyQT5與Candy風(fēng)格遷移模型實(shí)現(xiàn)一個可視化的風(fēng)格遷移界面程序，實(shí)現(xiàn)對任意視頻與圖像的Candy風(fēng)格化。這里我選取了一個輕量化的風(fēng)格遷移模型，整個模型大小只有不到2MB，模型的輸入與輸出格式與解釋截圖如下：

記住輸入圖像的通道順序是BGR、大小是720x720，然后輸出跟輸入的格式一致，但是輸出是浮點(diǎn)數(shù)，需要轉(zhuǎn)化一下才可以正常顯示。

運(yùn)行與演示 圖像輸入與風(fēng)格遷移效果：

視頻運(yùn)行

應(yīng)用程序演示

最終調(diào)用應(yīng)用程序代碼，實(shí)現(xiàn)啟動與運(yùn)行的界面如下

#初始化APP實(shí)例
importplatform
app=QtWidgets.QApplication(sys.argv)
if'Windows'==platform.system():
app.setStyle('Windows')
#初始化桌面容器
main_win=QtWidgets.QMainWindow()
#設(shè)置APP窗口名稱
main_win.setWindowTitle("圖像風(fēng)格遷移演示-2號高手")
#初始化內(nèi)容面板
content_panel=YOLOv8InferPanel()
#設(shè)置窗口大小
main_win.setMinimumSize(1280,860)
main_win.setCentralWidget(content_panel)

#請求顯示
main_win.show()
#加載窗口并啟動App
app.exec()

審核編輯：湯梓紅