2023年一月份跟二月份創建了一個PyQT5人工智能軟件開發系列的文章系列,過去的兩個月都沒怎么更新,心里一直想有時間繼續更新下去,今天又更新了一篇,基于PyQT5實現多線程、界面化、風格遷移模型的實時推理。
基本設計思路
這個系列我好久沒有更新了,今天更新一篇PyQT5中如何實現風格遷移模型的推理界面化與多線程支持。
參數設置選擇模型文件與數據文件地址作為輸入。 推理按鈕開發推理演示,支持視頻與圖像文件,開始推理會單獨開啟一個推理線程實現推理,推理結果通過信號發送到指定的槽函數處理之后更新界面,通過信號與槽機制實現界面線程跟推理線程相互獨立與數據共享。
模型說明
基于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()
審核編輯:湯梓紅
-
人工智能
+關注
關注
1791文章
46896瀏覽量
237664 -
多線程
+關注
關注
0文章
277瀏覽量
19923 -
模型
+關注
關注
1文章
3178瀏覽量
48730 -
pyqt5
+關注
關注
0文章
25瀏覽量
3394
原文標題:PyQT5+ONNXRUNTIME實現風格遷移應用
文章出處:【微信號:CVSCHOOL,微信公眾號:OpenCV學堂】歡迎添加關注!文章轉載請注明出處。
發布評論請先 登錄
相關推薦
評論