Compare commits
2 Commits
3d7c7a06e4
...
658560c34f
Author | SHA1 | Date | |
---|---|---|---|
658560c34f | |||
a41a4a2236 |
9
main.py
9
main.py
@ -10,7 +10,10 @@ 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 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):
|
class CameraThread(QThread):
|
||||||
"""摄像头线程类"""
|
"""摄像头线程类"""
|
||||||
@ -160,6 +163,10 @@ class MainWindow(QMainWindow):
|
|||||||
self.init_detector()
|
self.init_detector()
|
||||||
self.init_camera()
|
self.init_camera()
|
||||||
|
|
||||||
|
# 初始化OCR/CRNN模型(具体用哪个模块识别车牌号就写在这儿)
|
||||||
|
initialize_ocr_model()
|
||||||
|
|
||||||
|
|
||||||
def init_ui(self):
|
def init_ui(self):
|
||||||
"""初始化用户界面"""
|
"""初始化用户界面"""
|
||||||
self.setWindowTitle("车牌识别系统")
|
self.setWindowTitle("车牌识别系统")
|
||||||
|
Loading…
x
Reference in New Issue
Block a user