From a41a4a2236abcf639c30a64778f708324ee9f4dd Mon Sep 17 00:00:00 2001 From: spdis Date: Sat, 30 Aug 2025 12:23:01 +0800 Subject: [PATCH] yolorestart --- main.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/main.py b/main.py index b5b95c8..167de6a 100644 --- a/main.py +++ b/main.py @@ -10,7 +10,10 @@ from PyQt5.QtGui import QImage, QPixmap, QFont, QPainter, QPen, QColor import os from yolopart.detector import LicensePlateYOLO from OCR_part.ocr_interface import ocr_predict -#from CRNN_part.crnn_interface import crnn_predict（不使用CRNN） +from OCR_part.ocr_interface import initialize_ocr_model +#不使用CRNN所以注释掉 +#from CRNN_part.crnn_interface import crnn_predict +#from CRNN_part.crnn_interface import initialize_crnn_model class CameraThread(QThread): """摄像头线程类""" @@ -159,6 +162,10 @@ class MainWindow(QMainWindow): self.init_ui() self.init_detector() self.init_camera() + + # 初始化OCR/CRNN模型（具体用哪个模块识别车牌号就写在这儿） + initialize_ocr_model() + def init_ui(self): """初始化用户界面""" -- 2.47.2