YOLOv8+PyQT5打造細(xì)胞計(jì)數(shù)與識(shí)別應(yīng)用說(shuō)明

應(yīng)用說(shuō)明

YOLOv8對(duì)象檢測(cè)模型基于自定義數(shù)據(jù)集訓(xùn)練紅白細(xì)胞檢測(cè)模型，然后通過(guò)工具導(dǎo)出模型為ONNX，基于OpenVINO實(shí)現(xiàn)模型推理，完成細(xì)胞檢測(cè)識(shí)別，根據(jù)檢測(cè)到的細(xì)胞類別與數(shù)目，統(tǒng)計(jì)，在PyQT5打造的界面上顯示輸出檢測(cè)結(jié)果。

模型介紹

YOLOv8自定義紅白細(xì)胞檢測(cè)模型 模型的輸入與輸出信息如下：

輸入格式：

NCHW = 1x3x640x640

輸出格式:

1x6x8400， xyhw+類別2 = 6，20x20+40x40+80x80=8400

模型推理基于OpenVINO框架，CPU即可運(yùn)行。

界面設(shè)計(jì)

界面支持模型相關(guān)得參數(shù)設(shè)置、輸入圖像路徑選擇與設(shè)置，相關(guān)界面設(shè)計(jì)如下：

代碼實(shí)現(xiàn)與演示

代碼實(shí)現(xiàn)與演示，設(shè)計(jì)兩個(gè)類，一個(gè)界面類實(shí)現(xiàn)參數(shù)化界面輸入與推理結(jié)果顯示，另外一個(gè)模型推理類線程，負(fù)責(zé)調(diào)用模型實(shí)現(xiàn)模型推理，返回推理結(jié)果。

推理類與界面類之間通過(guò)信號(hào)與槽機(jī)制實(shí)現(xiàn)數(shù)據(jù)交換。

主界面調(diào)用界面類實(shí)例化與初始化，實(shí)現(xiàn)程序啟動(dòng)，在子線程種實(shí)現(xiàn)推理，主線程種刷新界面。

完整的主程序調(diào)用代碼實(shí)現(xià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("YOLOv8細(xì)胞計(jì)數(shù)與面積統(tǒng)計(jì)測(cè)量-演示版本")
#初始化內(nèi)容面板
content_panel=YOLOv8InferPanel()
#設(shè)置窗口大小
main_win.setMinimumSize(1340,910)
main_win.setCentralWidget(content_panel)

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

審核編輯：劉清