修改了模块的函数名,现在想用哪个模块直接导入即可

This commit is contained in:
2025-09-01 15:33:05 +08:00
parent 1c914cf89f
commit f342d37d63
3 changed files with 13 additions and 15 deletions

20
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
import os
from yolopart.detector import LicensePlateYOLO
from OCR_part.ocr_interface import ocr_predict
from OCR_part.ocr_interface import initialize_ocr_model
from OCR_part.ocr_interface import LPRNmodel_predict
from OCR_part.ocr_interface import LPRNinitialize_model
# 使用CRNN进行车牌字符识别可选同时也要修改第395396行
# from CRNN_part.crnn_interface import crnn_predict
# from CRNN_part.crnn_interface import initialize_crnn_model
# from CRNN_part.crnn_interface import LPRNmodel_predict
# from CRNN_part.crnn_interface import LPRNinitialize_model
class CameraThread(QThread):
"""摄像头线程类"""
@@ -163,9 +163,8 @@ class MainWindow(QMainWindow):
self.init_detector()
self.init_camera()
# 初始化OCR/CRNN模型具体用哪个模块识别车牌号就写在这儿
initialize_ocr_model()
# initialize_crnn_model()
# 初始化OCR/CRNN模型函数名改成一样的了,所以不要修改这里了,想用哪个模块直接导入
LPRNinitialize_model()
def init_ui(self):
@@ -390,10 +389,9 @@ class MainWindow(QMainWindow):
return "识别失败"
try:
# 使用OCR接口进行识别
# 可以根据需要切换为CRNN: crnn_predict(corrected_image)
result = ocr_predict(corrected_image)
# result = crnn_predict(corrected_image)
# 预测函数(来自模块)
# 函数名改成一样的了,所以不要修改这里了,想用哪个模块直接导入
result = LPRNmodel_predict(corrected_image)
# 将字符列表转换为字符串
if isinstance(result, list) and len(result) >= 7: