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): """初始化用户界面"""