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

基本設計思路

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

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

模型說明

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

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

運行與演示 圖像輸入與風格遷移效果：

視頻運行

應用程序演示

最終調用應用程序代碼，實現啟動與運行的界面如下

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

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

審核編輯：湯梓紅