本專案以 YOLOV8 加上 PySide6 視窗程式,即時偵測照片中的物件。請先下載 mainwindow.ui,置於專案下的 ui 目錄。
安裝套件
請先安裝如下套件
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 --no-cache-dir pip install pyside6 ultralytics opencv-python
ModelThread
載入模型可能需要一段時間,所以使用新執行緒載入。
from PySide6.QtCore import QThread, Signal
from ultralytics import YOLO
class ModelThread(QThread):
callback = Signal(object)
def __init__(self, parent=None):
super().__init__(parent)
def run(self):
model=YOLO("./yolov8n.pt")
self.callback.emit(model)
PictureThread
此段是載入檔案中的圖片,如果檔案很多,載入時間也會很久,所以也交由新執行緒載入。
