🍅大家好,今天给大家分享一个Python项目,感兴趣的可以先收藏起来,点赞、关注不迷路!🍅
大家在毕设选题,项目以及论文编写等相关问题都可以给我留言咨询,希望帮助同学们顺利毕业 。1、项目介绍
环境: Python 3.8 深度学习pytorch PyQt5图形化界面 LPRNet车牌识别算法 YOLOv8
车牌OCR识别模型:LPRNet
数据集:CCPD2019、CCPD2020数据集
其中资源包括:python、pycharm、cuda、torch等
技术栈: 本车牌识别系统是基于深度学习的,使用yolox进行车牌检测,然后通过LPRnet进行车牌识别。车牌检测模型目前包含yolox和yolov8,而且yolov8中包含个人修改的添加注意力机制的yolov8
2、项目界面
(1)上传图片进行车牌检测识别
(2)上传视频进行车牌检测识别
(3)连接摄像头进行车牌检测识别
(4)批量图片检测识别
(5)车牌检测识别记录
(6)车牌检测识别检测识别记录导出
3、项目说明
环境: Python 3.8 深度学习pytorch PyQt5图形化界面 LPRNet车牌识别算法 YOLOv8
技术栈: 本车牌识别系统是基于深度学习的,使用yolox进行车牌检测,然后通过LPRnet进行车牌识别。
车牌检测模型目前包含yolox和yolov8,而且yolov8中包含个人修改的添加注意力机制的yolov8
车牌OCR识别模型:LPRNet
数据集:CCPD2019、CCPD2020数据集
其中资源包括:python、pycharm、cuda、torch等
4、核心代码
from PyQt5.QtWidgets import QApplication, QMainWindow
from UI import Ui_MainWindow
import sys
from datetime import datetime
from PyQt5 import QtCore, QtGui, QtWidgets
from PyQt5.QtWidgets import QFileDialog, QMessageBox, QScrollBar
from PyQt5.QtGui import QBrush, QColor, QFont, QDesktopServices
from PyQt5.QtCore import QThread, pyqtSignal
from PyQt5.QtCore import Qt
# 添加yolox目标检测路径
import time
import os
import torch
from torch.autograd import Variable
sys.path.append('LicensePlate_detect/yolox_pytorch')
from LicensePlate_OCR.lpr_net.models.LPRNet import LPRNet, CHARS
from configs import config
import cv2
import sys, os, xlwt
import numpy as np
from PIL import Image
import threading
from LicensePlate_detect.yolox_pytorch.yolo import YOLOX_infer
from LicensePlate_detect.ultralytics.inferer import YOLOV8_infer
class MyMainWindow(QMainWindow, Ui_MainWindow):
def __init__(self):
super().__init__()
self.setupUi(self)
self.RowLength = 0
self.Data = [['序号', '图片名称', '录入时间', '车牌号码', '车牌类型', '车牌位置', '置信度', '车牌信息']]
self.number = 1
self.img_dirs_name = None
self.start_type = None
self.img = None
self.video = None
self.dir = None
self.img_path = None
self.img_path_dir = None
self.video_path = None
self.cam = None
self.output_dir = './output'
if not os.path.exists(self.output_dir):
os.mkdir(self.output_dir)
# 绑定退出程序
self.toolButton_exit.clicked.connect(QApplication.quit)
# 绑定打开文件夹
self.pushButton_dir.clicked.connect(self.open_dir)
# 打开视频
self.pushButton_video.clicked.connect(self.open_video)
# 打开cam
self.pushButton_cam.clicked.connect(self.open_cam)
# 绑定打开文件
self.pushButton_img.clicked.connect(self.open_dir_file)
# 绑定开始运行
self.pushButton_start.clicked.connect(self.start)
# 导出数据
self.pushButton_export.clicked.connect(self.writeFiles)
# 表格点击事件绑定
self.tableWidget_info.cellClicked.connect(self.cell_clicked)
# 连接单元格点击事件
def cell_clicked(self, row, column):
# 检查是否点击的是第一列(序号)
# if column:
# 获取第二列(图片名称)下的路径
image_name_item = self.tableWidget_info.item(row, 1)
if image_name_item is not None:
image_path = image_name_item.text()
img = cv2.imdecode(np.fromfile(image_path, dtype=np.uint8), -1)
self.show_frame(img)
def show_frame(self, img):
if img is not None:
# 尺寸适配
size = img.shape
if size[0] / size[1] > 1.0907:
w = size