spdis/License_plate_recognition
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

更新 main.py

Browse Source
This commit is contained in:
spdis 2025-08-31 12:19:25 +08:00
parent 7f89965956
commit afe15b990a
1 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
main.py
View File

@ -9,11 +9,11 @@ from PyQt5.QtCore import QTimer, Qt, pyqtSignal, QThread
from PyQt5.QtGui import QImage, QPixmap, QFont, QPainter, QPen, QColor from PyQt5.QtGui import QImage, QPixmap, QFont, QPainter, QPen, QColor
import os import os
from yolopart.detector import LicensePlateYOLO from yolopart.detector import LicensePlateYOLO
from OCR_part.ocr_interface import ocr_predict #from OCR_part.ocr_interface import ocr_predict
from OCR_part.ocr_interface import initialize_ocr_model #from OCR_part.ocr_interface import initialize_ocr_model
#不使用CRNN所以注释掉 # 使用CRNN进行车牌字符识别
#from CRNN_part.crnn_interface import crnn_predict from CRNN_part.crnn_interface import crnn_predict
#from CRNN_part.crnn_interface import initialize_crnn_model from CRNN_part.crnn_interface import initialize_crnn_model
class CameraThread(QThread): class CameraThread(QThread):
"""摄像头线程类""" """摄像头线程类"""
@ -164,7 +164,8 @@ class MainWindow(QMainWindow):
self.init_camera() self.init_camera()
# 初始化OCR/CRNN模型（具体用哪个模块识别车牌号就写在这儿） # 初始化OCR/CRNN模型（具体用哪个模块识别车牌号就写在这儿）
initialize_ocr_model() #initialize_ocr_model()
initialize_crnn_model()
def init_ui(self): def init_ui(self):
@ -391,7 +392,8 @@ class MainWindow(QMainWindow):
try: try:
# 使用OCR接口进行识别 # 使用OCR接口进行识别
# 可以根据需要切换为CRNN: crnn_predict(corrected_image) # 可以根据需要切换为CRNN: crnn_predict(corrected_image)
result = ocr_predict(corrected_image) #result = ocr_predict(corrected_image)
result = crnn_predict(corrected_image)
# 将字符列表转换为字符串 # 将字符列表转换为字符串
if isinstance(result, list) and len(result) >= 7: if isinstance(result, list) and len(result) >= 7: