基于YOLOv8深度学习+Pyqt5的电动车头盔佩戴检测系统

wx供重浩：创享日记
对话框发送：225头盔
获取完整源码源文件+已标注的数据集（1463张）+源码各文件说明+配置跑通说明文档
若需要一对一远程操作在你电脑跑通，有偿59yuan

效果展示

基于YOLOv8深度学习+PyQT5的电动车头盔佩戴检测系统

在这里插入图片描述

各文件说明

在这里插入图片描述

随着电动车的普及，交通安全问题日益凸显。头盔作为保护骑行者头部安全的重要装备，其佩戴情况的监测对于减少交通事故伤害具有重要意义。本文提出了一种基于YOLOv8（You Only Look Once version 8）的电动车头盔佩戴检测系统，旨在实时监测并提醒骑行者佩戴头盔。该系统利用深度学习技术，通过训练YOLOv8模型来识别电动车骑行者是否佩戴头盔，并在检测到未佩戴头盔的情况下发出警报。系统设计考虑了实时性、准确性和用户友好性，为提升电动车骑行安全提供了一种有效的技术解决方案。

电动车作为一种便捷的交通工具，在城市交通中扮演着越来越重要的角色。然而，由于缺乏必要的安全措施，电动车事故频发，尤其是头部伤害，往往导致严重后果。头盔的正确佩戴可以有效降低头部受伤的风险。因此，开发一种能够实时监测头盔佩戴情况的系统，对于提高骑行者的安全意识和减少交通事故具有重要作用。

YOLOv8是YOLO系列目标检测模型的最新版本，它在前代模型的基础上进行了优化，提高了检测速度和准确性。YOLOv8采用了多尺度特征图的融合、更精细的锚点框设计以及更有效的损失函数，使得模型在处理复杂场景时表现出色。

系统设计
数据集准备
为了训练YOLOv8模型，首先需要收集和标注一个包含大量电动车骑行者图像的数据集。数据集应包含各种天气条件、不同光照环境以及多种头盔类型。图像标注需要指出头盔的位置和类别，以便模型学习区分佩戴头盔与否。

模型训练
使用准备好的数据集，对YOLOv8模型进行训练。训练过程中，需要调整学习率、批大小、训练周期等超参数，以获得最佳的检测效果。同时，可以采用数据增强技术，如随机裁剪、旋转、颜色变换等，以提高模型的泛化能力。

系统实现
系统的核心是YOLOv8模型，它负责从输入的实时视频流中检测头盔。系统还包括一个用户界面，用于显示检测结果和发出警报。当模型检测到未佩戴头盔的骑行者时，系统会通过声音或视觉信号提醒用户。

实时检测
为了实现实时检测，系统需要能够在低延迟的情况下处理视频流。这要求模型不仅要准确，还要高效。在实际部署中，可能需要在边缘设备上进行模型推理，以减少对中心服务器的依赖。

结果与讨论
在测试集上，YOLOv8模型展现出了较高的头盔检测准确率。在不同的场景和光照条件下，模型均能稳定地识别出佩戴和未佩戴头盔的骑行者。然而，模型在处理遮挡严重或头盔与骑行者头部颜色相近的情况下仍有改进空间。

结论
本文提出的基于YOLOv8的电动车头盔佩戴检测系统，能够有效地提高电动车骑行者的安全意识。通过实时监测和提醒，该系统有助于减少因未佩戴头盔导致的交通事故伤害。未来的工作将集中在进一步提高模型的鲁棒性，以及探索更高效的模型部署方案。

软件系统部分源码

# -*- coding: utf-8 -*-
import time
from PyQt5.QtWidgets import QApplication , QMainWindow, QFileDialog, \QMessageBox,QWidget,QHeaderView,QTableWidgetItem, QAbstractItemView
import sys
import os
from PIL import ImageFont
from ultralytics import YOLO
sys.path.append('UIProgram')
from UIProgram.UiMain import Ui_MainWindow
import sys
from PyQt5.QtCore import QTimer, Qt, QThread, pyqtSignal,QCoreApplication
import detect_tools as tools
import cv2
import Config
from UIProgram.QssLoader import QSSLoader
from UIProgram.precess_bar import ProgressBar
import numpy as np
# import torchclass MainWindow(QMainWindow):def __init__(self, parent=None):super(QMainWindow, self).__init__(parent)self.ui = Ui_MainWindow()self.ui.setupUi(self)self.initMain()self.signalconnect()# 加载css渲染效果style_file = 'UIProgram/style.css'qssStyleSheet = QSSLoader.read_qss_file(style_file)self.setStyleSheet(qssStyleSheet)def signalconnect(self):self.ui.PicBtn.clicked.connect(self.open_img)self.ui.comboBox.activated.connect(self.combox_change)self.ui.VideoBtn.clicked.connect(self.vedio_show)self.ui.CapBtn.clicked.connect(self.camera_show)self.ui.SaveBtn.clicked.connect(self.save_detect_video)self.ui.ExitBtn.clicked.connect(QCoreApplication.quit)self.ui.FilesBtn.clicked.connect(self.detact_batch_imgs)def initMain(self):self.show_width = 770self.show_height = 480self.org_path = Noneself.is_camera_open = Falseself.cap = None# self.device = 0 if torch.cuda.is_available() else 'cpu'# 加载检测模型self.model = YOLO(Config.model_path, task='detect')self.model(np.zeros((48, 48, 3)))  #预先加载推理模型self.fontC = ImageFont.truetype("Font/platech.ttf", 25, 0)# 用于绘制不同颜色矩形框self.colors = tools.Colors()# 更新视频图像self.timer_camera = QTimer()# 更新检测信息表格# self.timer_info = QTimer()# 保存视频self.timer_save_video = QTimer()# 表格self.ui.tableWidget.verticalHeader().setSectionResizeMode(QHeaderView.Fixed)self.ui.tableWidget.verticalHeader().setDefaultSectionSize(40)self.ui.tableWidget.setColumnWidth(0, 80)  # 设置列宽self.ui.tableWidget.setColumnWidth(1, 200)self.ui.tableWidget.setColumnWidth(2, 150)self.ui.tableWidget.setColumnWidth(3, 90)self.ui.tableWidget.setColumnWidth(4, 230)# self.ui.tableWidget.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)  # 表格铺满# self.ui.tableWidget.horizontalHeader().setSectionResizeMode(0, QHeaderView.Interactive)# self.ui.tableWidget.setEditTriggers(QAbstractItemView.NoEditTriggers)  # 设置表格不可编辑self.ui.tableWidget.setSelectionBehavior(QAbstractItemView.SelectRows)  # 设置表格整行选中self.ui.tableWidget.verticalHeader().setVisible(False)  # 隐藏列标题self.ui.tableWidget.setAlternatingRowColors(True)  # 表格背景交替# 设置主页背景图片border-image: url(:/icons/ui_imgs/icons/camera.png)# self.setStyleSheet("#MainWindow{background-image:url(:/bgs/ui_imgs/bg3.jpg)}")def open_img(self):if self.cap:# 打开图片前关闭摄像头self.video_stop()self.is_camera_open = Falseself.ui.CaplineEdit.setText('摄像头未开启')self.cap = None# 弹出的窗口名称：'打开图片'# 默认打开的目录：'./'# 只能打开.jpg与.gif结尾的图片文件# file_path, _ = QFileDialog.getOpenFileName(self.ui.centralwidget, '打开图片', './', "Image files (*.jpg *.gif)")file_path, _ = QFileDialog.getOpenFileName(None, '打开图片', './', "Image files (*.jpg *.jepg *.png)")if not file_path:returnself.ui.comboBox.setDisabled(False)self.org_path = file_pathself.org_img = tools.img_cvread(self.org_path)# 目标检测t1 = time.time()self.results = self.model(self.org_path)[0]t2 = time.time()take_time_str = '{:.3f} s'.format(t2 - t1)self.ui.time_lb.setText(take_time_str)location_list = self.results.boxes.xyxy.tolist()self.location_list = [list(map(int, e)) for e in location_list]cls_list = self.results.boxes.cls.tolist()self.cls_list = [int(i) for i in cls_list]self.conf_list = self.results.boxes.conf.tolist()self.conf_list = ['%.2f %%' % (each*100) for each in self.conf_list]# now_img = self.cv_img.copy()# for loacation, type_id, conf in zip(self.location_list, self.cls_list, self.conf_list):#     type_id = int(type_id)#     color = self.colors(int(type_id), True)#     # cv2.rectangle(now_img, (int(x1), int(y1)), (int(x2), int(y2)), colors(int(type_id), True), 3)#     now_img = tools.drawRectBox(now_img, loacation, Config.CH_names[type_id], self.fontC, color)now_img = self.results.plot()self.draw_img = now_img# 获取缩放后的图片尺寸self.img_width, self.img_height = self.get_resize_size(now_img)resize_cvimg = cv2.resize(now_img,(self.img_width, self.img_height))pix_img = tools.cvimg_to_qpiximg(resize_cvimg)self.ui.label_show.setPixmap(pix_img)self.ui.label_show.setAlignment(Qt.AlignCenter)# 设置路径显示self.ui.PiclineEdit.setText(self.org_path)# 目标数目target_nums = len(self.cls_list)self.ui.label_nums.setText(str(target_nums))# 设置目标选择下拉框choose_list = ['全部']target_names = [Config.names[id]+ '_'+ str(index) for index,id in enumerate(self.cls_list)]# object_list = sorted(set(self.cls_list))# for each in object_list:#     choose_list.append(Config.CH_names[each])choose_list = choose_list + target_namesself.ui.comboBox.clear()self.ui.comboBox.addItems(choose_list)if target_nums >= 1:self.ui.type_lb.setText(Config.CH_names[self.cls_list[0]])self.ui.label_conf.setText(str(self.conf_list[0]))#   默认显示第一个目标框坐标#   设置坐标位置值self.ui.label_xmin.setText(str(self.location_list[0][0]))self.ui.label_ymin.setText(str(self.location_list[0][1]))self.ui.label_xmax.setText(str(self.location_list[0][2]))self.ui.label_ymax.setText(str(self.location_list[0][3]))else:self.ui.type_lb.setText('')self.ui.label_conf.setText('')self.ui.label_xmin.setText('')self.ui.label_ymin.setText('')self.ui.label_xmax.setText('')self.ui.label_ymax.setText('')# # 删除表格所有行self.ui.tableWidget.setRowCount(0)self.ui.tableWidget.clearContents()self.tabel_info_show(self.location_list, self.cls_list, self.conf_list,path=self.org_path)def detact_batch_imgs(self):if self.cap:# 打开图片前关闭摄像头self.video_stop()self.is_camera_open = Falseself.ui.CaplineEdit.setText('摄像头未开启')self.cap = Nonedirectory = QFileDialog.getExistingDirectory(self,"选取文件夹","./")  # 起始路径if not  directory:returnself.org_path = directoryimg_suffix = ['jpg','png','jpeg','bmp']for file_name in os.listdir(directory):full_path = os.path.join(directory,file_name)if os.path.isfile(full_path) and file_name.split('.')[-1].lower() in img_suffix:# self.ui.comboBox.setDisabled(False)img_path = full_pathself.org_img = tools.img_cvread(img_path)# 目标检测t1 = time.time()self.results = self.model(img_path)[0]t2 = time.time()take_time_str = '{:.3f} s'.format(t2 - t1)self.ui.time_lb.setText(take_time_str)location_list = self.results.boxes.xyxy.tolist()self.location_list = [list(map(int, e)) for e in location_list]cls_list = self.results.boxes.cls.tolist()self.cls_list = [int(i) for i in cls_list]self.conf_list = self.results.boxes.conf.tolist()self.conf_list = ['%.2f %%' % (each * 100) for each in self.conf_list]now_img = self.results.plot()self.draw_img = now_img# 获取缩放后的图片尺寸self.img_width, self.img_height = self.get_resize_size(now_img)resize_cvimg = cv2.resize(now_img, (self.img_width, self.img_height))pix_img = tools.cvimg_to_qpiximg(resize_cvimg)self.ui.label_show.setPixmap(pix_img)self.ui.label_show.setAlignment(Qt.AlignCenter)# 设置路径显示self.ui.PiclineEdit.setText(img_path)# 目标数目target_nums = len(self.cls_list)self.ui.label_nums.setText(str(target_nums))# 设置目标选择下拉框choose_list = ['全部']target_names = [Config.names[id] + '_' + str(index) for index, id in enumerate(self.cls_list)]choose_list = choose_list + target_namesself.ui.comboBox.clear()self.ui.comboBox.addItems(choose_list)if target_nums >= 1:self.ui.type_lb.setText(Config.CH_names[self.cls_list[0]])self.ui.label_conf.setText(str(self.conf_list[0]))#   默认显示第一个目标框坐标#   设置坐标位置值self.ui.label_xmin.setText(str(self.location_list[0][0]))self.ui.label_ymin.setText(str(self.location_list[0][1]))self.ui.label_xmax.setText(str(self.location_list[0][2]))self.ui.label_ymax.setText(str(self.location_list[0][3]))else:self.ui.type_lb.setText('')self.ui.label_conf.setText('')self.ui.label_xmin.setText('')self.ui.label_ymin.setText('')self.ui.label_xmax.setText('')self.ui.label_ymax.setText('')# # 删除表格所有行# self.ui.tableWidget.setRowCount(0)# self.ui.tableWidget.clearContents()self.tabel_info_show(self.location_list, self.cls_list, self.conf_list, path=img_path)self.ui.tableWidget.scrollToBottom()QApplication.processEvents()  #刷新页面def draw_rect_and_tabel(self, results, img):now_img = img.copy()location_list = results.boxes.xyxy.tolist()self.location_list = [list(map(int, e)) for e in location_list]cls_list = results.boxes.cls.tolist()self.cls_list = [int(i) for i in cls_list]self.conf_list = results.boxes.conf.tolist()self.conf_list = ['%.2f %%' % (each * 100) for each in self.conf_list]for loacation, type_id, conf in zip(self.location_list, self.cls_list, self.conf_list):type_id = int(type_id)color = self.colors(int(type_id), True)# cv2.rectangle(now_img, (int(x1), int(y1)), (int(x2), int(y2)), colors(int(type_id), True), 3)now_img = tools.drawRectBox(now_img, loacation, Config.CH_names[type_id], self.fontC, color)# 获取缩放后的图片尺寸self.img_width, self.img_height = self.get_resize_size(now_img)resize_cvimg = cv2.resize(now_img, (self.img_width, self.img_height))pix_img = tools.cvimg_to_qpiximg(resize_cvimg)self.ui.label_show.setPixmap(pix_img)self.ui.label_show.setAlignment(Qt.AlignCenter)# 设置路径显示self.ui.PiclineEdit.setText(self.org_path)# 目标数目target_nums = len(self.cls_list)self.ui.label_nums.setText(str(target_nums))if target_nums >= 1:self.ui.type_lb.setText(Config.CH_names[self.cls_list[0]])self.ui.label_conf.setText(str(self.conf_list[0]))self.ui.label_xmin.setText(str(self.location_list[0][0]))self.ui.label_ymin.setText(str(self.location_list[0][1]))self.ui.label_xmax.setText(str(self.location_list[0][2]))self.ui.label_ymax.setText(str(self.location_list[0][3]))else:self.ui.type_lb.setText('')self.ui.label_conf.setText('')self.ui.label_xmin.setText('')self.ui.label_ymin.setText('')self.ui.label_xmax.setText('')self.ui.label_ymax.setText('')# 删除表格所有行self.ui.tableWidget.setRowCount(0)self.ui.tableWidget.clearContents()self.tabel_info_show(self.location_list, self.cls_list, self.conf_list, path=self.org_path)return now_imgdef combox_change(self):com_text = self.ui.comboBox.currentText()if com_text == '全部':cur_box = self.location_listcur_img = self.results.plot()self.ui.type_lb.setText(Config.CH_names[self.cls_list[0]])self.ui.label_conf.setText(str(self.conf_list[0]))else:index = int(com_text.split('_')[-1])cur_box = [self.location_list[index]]cur_img = self.results[index].plot()self.ui.type_lb.setText(Config.CH_names[self.cls_list[index]])self.ui.label_conf.setText(str(self.conf_list[index]))# 设置坐标位置值self.ui.label_xmin.setText(str(cur_box[0][0]))self.ui.label_ymin.setText(str(cur_box[0][1]))self.ui.label_xmax.setText(str(cur_box[0][2]))self.ui.label_ymax.setText(str(cur_box[0][3]))resize_cvimg = cv2.resize(cur_img, (self.img_width, self.img_height))pix_img = tools.cvimg_to_qpiximg(resize_cvimg)self.ui.label_show.clear()self.ui.label_show.setPixmap(pix_img)self.ui.label_show.setAlignment(Qt.AlignCenter)def get_video_path(self):file_path, _ = QFileDialog.getOpenFileName(None, '打开视频', './', "Image files (*.avi *.mp4 *.jepg *.png)")if not file_path:return Noneself.org_path = file_pathself.ui.VideolineEdit.setText(file_path)return file_pathdef video_start(self):# 删除表格所有行self.ui.tableWidget.setRowCount(0)self.ui.tableWidget.clearContents()# 清空下拉框self.ui.comboBox.clear()# 定时器开启，每隔一段时间，读取一帧self.timer_camera.start(1)self.timer_camera.timeout.connect(self.open_frame)def tabel_info_show(self, locations, clses, confs, path=None):path = pathfor location, cls, conf in zip(locations, clses, confs):row_count = self.ui.tableWidget.rowCount()  # 返回当前行数(尾部)self.ui.tableWidget.insertRow(row_count)  # 尾部插入一行item_id = QTableWidgetItem(str(row_count+1))  # 序号item_id.setTextAlignment(Qt.AlignHCenter | Qt.AlignVCenter)  # 设置文本居中item_path = QTableWidgetItem(str(path))  # 路径# item_path.setTextAlignment(Qt.AlignHCenter | Qt.AlignVCenter)item_cls = QTableWidgetItem(str(Config.CH_names[cls]))item_cls.setTextAlignment(Qt.AlignHCenter | Qt.AlignVCenter)  # 设置文本居中item_conf = QTableWidgetItem(str(conf))item_conf.setTextAlignment(Qt.AlignHCenter | Qt.AlignVCenter)  # 设置文本居中item_location = QTableWidgetItem(str(location)) # 目标框位置# item_location.setTextAlignment(Qt.AlignHCenter | Qt.AlignVCenter)  # 设置文本居中self.ui.tableWidget.setItem(row_count, 0, item_id)self.ui.tableWidget.setItem(row_count, 1, item_path)self.ui.tableWidget.setItem(row_count, 2, item_cls)self.ui.tableWidget.setItem(row_count, 3, item_conf)self.ui.tableWidget.setItem(row_count, 4, item_location)self.ui.tableWidget.scrollToBottom()def video_stop(self):self.cap.release()self.timer_camera.stop()# self.timer_info.stop()def open_frame(self):ret, now_img = self.cap.read()if ret:# 目标检测t1 = time.time()results = self.model(now_img)[0]t2 = time.time()take_time_str = '{:.3f} s'.format(t2 - t1)self.ui.time_lb.setText(take_time_str)location_list = results.boxes.xyxy.tolist()self.location_list = [list(map(int, e)) for e in location_list]cls_list = results.boxes.cls.tolist()self.cls_list = [int(i) for i in cls_list]self.conf_list = results.boxes.conf.tolist()self.conf_list = ['%.2f %%' % (each * 100) for each in self.conf_list]now_img = results.plot()# 获取缩放后的图片尺寸self.img_width, self.img_height = self.get_resize_size(now_img)resize_cvimg = cv2.resize(now_img, (self.img_width, self.img_height))pix_img = tools.cvimg_to_qpiximg(resize_cvimg)self.ui.label_show.setPixmap(pix_img)self.ui.label_show.setAlignment(Qt.AlignCenter)# 目标数目target_nums = len(self.cls_list)self.ui.label_nums.setText(str(target_nums))# 设置目标选择下拉框choose_list = ['全部']target_names = [Config.names[id] + '_' + str(index) for index, id in enumerate(self.cls_list)]# object_list = sorted(set(self.cls_list))# for each in object_list:#     choose_list.append(Config.CH_names[each])choose_list = choose_list + target_namesself.ui.comboBox.clear()self.ui.comboBox.addItems(choose_list)if target_nums >= 1:self.ui.type_lb.setText(Config.CH_names[self.cls_list[0]])self.ui.label_conf.setText(str(self.conf_list[0]))#   默认显示第一个目标框坐标#   设置坐标位置值self.ui.label_xmin.setText(str(self.location_list[0][0]))self.ui.label_ymin.setText(str(self.location_list[0][1]))self.ui.label_xmax.setText(str(self.location_list[0][2]))self.ui.label_ymax.setText(str(self.location_list[0][3]))else:self.ui.type_lb.setText('')self.ui.label_conf.setText('')self.ui.label_xmin.setText('')self.ui.label_ymin.setText('')self.ui.label_xmax.setText('')self.ui.label_ymax.setText('')# 删除表格所有行# self.ui.tableWidget.setRowCount(0)# self.ui.tableWidget.clearContents()self.tabel_info_show(self.location_list, self.cls_list, self.conf_list, path=self.org_path)else:self.cap.release()self.timer_camera.stop()def vedio_show(self):if self.is_camera_open:self.is_camera_open = Falseself.ui.CaplineEdit.setText('摄像头未开启')video_path = self.get_video_path()if not video_path:return Noneself.cap = cv2.VideoCapture(video_path)self.video_start()self.ui.comboBox.setDisabled(True)def camera_show(self):self.is_camera_open = not self.is_camera_openif self.is_camera_open:self.ui.CaplineEdit.setText('摄像头开启')self.cap = cv2.VideoCapture(0)self.video_start()self.ui.comboBox.setDisabled(True)else:self.ui.CaplineEdit.setText('摄像头未开启')self.ui.label_show.setText('')if self.cap:self.cap.release()cv2.destroyAllWindows()self.ui.label_show.clear()def get_resize_size(self, img):_img = img.copy()img_height, img_width , depth= _img.shaperatio = img_width / img_heightif ratio >= self.show_width / self.show_height:self.img_width = self.show_widthself.img_height = int(self.img_width / ratio)else:self.img_height = self.show_heightself.img_width = int(self.img_height * ratio)return self.img_width, self.img_heightdef save_detect_video(self):if self.cap is None and not self.org_path:QMessageBox.about(self, '提示', '当前没有可保存信息，请先打开图片或视频！')returnif self.is_camera_open:QMessageBox.about(self, '提示', '摄像头视频无法保存!')returnif self.cap:res = QMessageBox.information(self, '提示', '保存视频检测结果可能需要较长时间，请确认是否继续保存？',QMessageBox.Yes | QMessageBox.No ,  QMessageBox.Yes)if res == QMessageBox.Yes:self.video_stop()com_text = self.ui.comboBox.currentText()self.btn2Thread_object = btn2Thread(self.org_path, self.model, com_text)self.btn2Thread_object.start()self.btn2Thread_object.update_ui_signal.connect(self.update_process_bar)else:returnelse:if os.path.isfile(self.org_path):fileName = os.path.basename(self.org_path)name , end_name= fileName.rsplit(".",1)save_name = name + '_detect_result.' + end_namesave_img_path = os.path.join(Config.save_path, save_name)# 保存图片cv2.imwrite(save_img_path, self.draw_img)QMessageBox.about(self, '提示', '图片保存成功!\n文件路径:{}'.format(save_img_path))else:img_suffix = ['jpg', 'png', 'jpeg', 'bmp']for file_name in os.listdir(self.org_path):full_path = os.path.join(self.org_path, file_name)if os.path.isfile(full_path) and file_name.split('.')[-1].lower() in img_suffix:name, end_name = file_name.rsplit(".",1)save_name = name + '_detect_result.' + end_namesave_img_path = os.path.join(Config.save_path, save_name)results = self.model(full_path)[0]now_img = results.plot()# 保存图片cv2.imwrite(save_img_path, now_img)QMessageBox.about(self, '提示', '图片保存成功!\n文件路径:{}'.format(Config.save_path))def update_process_bar(self,cur_num, total):if cur_num == 1:self.progress_bar = ProgressBar(self)self.progress_bar.show()if cur_num >= total:self.progress_bar.close()QMessageBox.about(self, '提示', '视频保存成功!\n文件在{}目录下'.format(Config.save_path))returnif self.progress_bar.isVisible() is False:# 点击取消保存时，终止进程self.btn2Thread_object.stop()returnvalue = int(cur_num / total *100)self.progress_bar.setValue(cur_num, total, value)QApplication.processEvents()class btn2Thread(QThread):"""进行检测后的视频保存"""# 声明一个信号update_ui_signal = pyqtSignal(int,int)def __init__(self, path, model, com_text):super(btn2Thread, self).__init__()self.org_path = pathself.model = modelself.com_text = com_text# 用于绘制不同颜色矩形框self.colors = tools.Colors()self.is_running = True  # 标志位，表示线程是否正在运行def run(self):# VideoCapture方法是cv2库提供的读取视频方法cap = cv2.VideoCapture(self.org_path)# 设置需要保存视频的格式“xvid”# 该参数是MPEG-4编码类型，文件名后缀为.avifourcc = cv2.VideoWriter_fourcc(*'XVID')# 设置视频帧频fps = cap.get(cv2.CAP_PROP_FPS)# 设置视频大小size = (int(cap.get(cv2.CAP_PROP_FRAME_WIDTH)), int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT)))# VideoWriter方法是cv2库提供的保存视频方法# 按照设置的格式来out输出fileName = os.path.basename(self.org_path)name, end_name = fileName.split('.')save_name = name + '_detect_result.avi'save_video_path = os.path.join(Config.save_path, save_name)out = cv2.VideoWriter(save_video_path, fourcc, fps, size)prop = cv2.CAP_PROP_FRAME_COUNTtotal = int(cap.get(prop))print("[INFO] 视频总帧数：{}".format(total))cur_num = 0# 确定视频打开并循环读取while (cap.isOpened() and self.is_running):cur_num += 1print('当前第{}帧，总帧数{}'.format(cur_num, total))# 逐帧读取，ret返回布尔值# 参数ret为True 或者False,代表有没有读取到图片# frame表示截取到一帧的图片ret, frame = cap.read()if ret == True:# 检测results = self.model(frame)[0]frame = results.plot()out.write(frame)self.update_ui_signal.emit(cur_num, total)else:break# 释放资源cap.release()out.release()def stop(self):self.is_running = Falseif __name__ == "__main__":app = QApplication(sys.argv)win = MainWindow()win.show()sys.exit(app.exec_())

本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如若转载，请注明出处：http://www.rhkb.cn/news/264877.html

如若内容造成侵权/违法违规/事实不符，请联系长河编程网进行投诉反馈email:809451989@qq.com，一经查实，立即删除！

相关文章

C语言第三十一弹---自定义类型：结构体(下)

C语言第三十一弹---自定义类型：结构体(下)

✨个人主页： 熬夜学编程的小林 💗系列专栏： 【C语言详解】 【数据结构详解】 目录 1、结构体内存对齐 1.1、为什么存在内存对齐? 1.2、修改默认对齐数 2、结构体传参 3、结构体实现位段 3.1、什么是位段 3.2、位段的内存分配 3.3、…
阅读更多...
论文精读--GPT3

论文精读--GPT3

不像GPT2一样追求zero-shot，而换成了few-shot Abstract Recent work has demonstrated substantial gains on many NLP tasks and benchmarks by pre-training on a large corpus of text followed by fine-tuning on a specific task. While typically task-agnos…
阅读更多...
EMQX Enterprise 5.5 发布：新增 Elasticsearch 数据集成

EMQX Enterprise 5.5 发布：新增 Elasticsearch 数据集成

EMQX Enterprise 5.5.0 版本已正式发布！ 在这个版本中，我们引入了一系列新的功能和改进，包括对 Elasticsearch 的集成、Apache IoTDB 和 OpenTSDB 数据集成优化、授权缓存支持排除主题等功能。此外，新版本还进行了多项改进以及 B…
阅读更多...
多模态表征—CLIP及中文版Chinese-CLIP：理论讲解、代码微调与论文阅读

多模态表征—CLIP及中文版Chinese-CLIP：理论讲解、代码微调与论文阅读

我之前一直在使用CLIP/Chinese-CLIP，但并未进行过系统的疏导。这次正好可以详细解释一下。相比于CLIP模型，Chinese-CLIP更适合我们的应用和微调，因为原始的CLIP模型只支持英文，对于我们的中文应用来说不够友好。Chinese-CLIP很好地…
阅读更多...
unity发布webGL压缩方式的gzip，使用nginx作为web服务器时的配置文件

unity发布webGL压缩方式的gzip，使用nginx作为web服务器时的配置文件

unity发布webGL压缩方式的gzip，使用nginx作为web服务器时的配置文件 Unity版本是：2021.3 nginx的版本是：nginx-1.25.4 Unity发布webgl时的测试 设置压缩方式是gzip nginx配置文件 worker_processes 1;events {worker_connections 102…
阅读更多...
Linux系统——Nginx小总结

Linux系统——Nginx小总结

目录 一、影响用户体验的因素 二、网络连接——Apache/Nginx服务请求过程 三、I/O模型——Input/Output模型 1.同步/异步 2.阻塞/非阻塞 3.同步异步/阻塞非阻塞组合 四、Nginx用法 一、影响用户体验的因素 客户端硬件配置客户端网络速率客户端与服务端距离服务端网络速…
阅读更多...
SQLlabs46关

SQLlabs46关

看看源码 最终我们的id是放到order by后面了 如果我们直接用列去排序 ?sortusername/password username： passward 可以看到顺序是不同的，当然第一列第二列第三列也可以，基本上都是这个原理，那怎么去实现注入呢，我…
阅读更多...
Rider 2023：打造高效.NET项目的智能IDE，让开发更简单mac/win版

Rider 2023：打造高效.NET项目的智能IDE，让开发更简单mac/win版

JetBrains Rider 2023激活版下载是一款专为.NET开发者打造的强大集成开发环境（IDE）。这款IDE提供了丰富的功能，旨在帮助开发者更快速、更高效地编写、调试和测试.NET应用程序。 Rider 2023 软件获取 Rider 2023在保持了其一贯的智能代码补全…
阅读更多...
Git笔记——3

Git笔记——3

提示：文章写完后，目录可以自动生成，如何生成可参考右边的帮助文档 文章目录 前言 一、合并模式和分支策略 二、bug分支 三、强制删除分支 四、创建远程仓库 五、克隆远程仓库_HTTPS和_SSH 克隆远程仓库_HTTPS 克隆远程仓库_SSH 六、向远程仓库…
阅读更多...
1.1_1 计算机网络的概念、功能、组成和分类

1.1_1 计算机网络的概念、功能、组成和分类

文章目录 1.1_1 计算机网络的概念、功能、组成和分类（一）计算机网络的概念（二）计算机网络的功能（三）计算机网络的组成1.组成部分2.工作方式3.功能组成 （四）计算机网络的分类 总结 1.…
阅读更多...
Qt QWidget 简约美观的加载动画 第五季 - 小方块风格

Qt QWidget 简约美观的加载动画 第五季 - 小方块风格

给大家分享两个小方块风格的加载动画 &#x1f60a; 第五季来啦 &#x1f60a; 效果如下: 一个三个文件,可以直接编译运行 //main.cpp #include "LoadingAnimWidget.h" #include <QApplication> #include <QGridLayout> int main(int argc, char *arg…
阅读更多...
sql-labs第46关 order by盲注

sql-labs第46关 order by盲注

sql-labs第46关 order by盲注 来到了第46关进入关卡发现让我们输入的参数为sort&#xff0c;我们输入?sort1尝试&#xff1a; 输入?sort2,3,发现表格按照顺序进行排列输出&#xff0c;明显是使用了order by相关的函数。 我们将参数变成1进行尝试&#xff0c;就会报错&…
阅读更多...
Rust之构建命令行程序(四)：用TDD(测试-驱动-开发)模式来开发库的功能

Rust之构建命令行程序(四)：用TDD(测试-驱动-开发)模式来开发库的功能

开发环境 Windows 11Rust 1.75.0 VS Code 1.86.2 项目工程 这次创建了新的工程minigrep. 用测试-驱动模式来开发库的功能 既然我们已经将逻辑提取到src/lib.rs中&#xff0c;并将参数收集和错误处理留在src/main.rs中&#xff0c;那么为代码的核心功能编写测试就容易多了。我…
阅读更多...
抽象的后端

抽象的后端

Connection refused: no further information 出现这条代码的核心是你使用redis&#xff0c;但是本地没有开启redis服务 如何启动redis服务 第一步&#xff1a;确定你安装了对应的框架 以spring为例 <dependency><groupId>org.springframework.boot</group…
阅读更多...
初识Lombok

初识Lombok

前言 最近读一些公司的业务代码&#xff0c;发现近几年的java项目工程中都使用了lombok&#xff0c;lombok是一个可以自动生成get,set、toString等模板类方法的工具框架&#xff0c;程序再引入lombok后&#xff0c;添加一个注解便可以不写get\set\toString等方法。 Lombok示例…
阅读更多...
【DL】深度学习之语音识别

【DL】深度学习之语音识别

目录 1 核心概念 2 安装依赖库 3 实践 语音信号处理&#xff08;Speech Signal Processing&#xff09;简称语音处理。 语音识别&#xff08;ASR&#xff09;和自然语言处理&#xff08;NLP&#xff09;&#xff1a;语音识别就是将语音信号转化成文字文本&#xff0c;简单实…
阅读更多...
UE5 C++ Widget练习 Button 和 ProgressBar创建血条

UE5 C++ Widget练习 Button 和 ProgressBar创建血条

一. 1.C创建一个继承Widget类的子类&#xff0c; 命名为MyUserWidget 2.加上Button 和 UserWidget的头文件 #include "CoreMinimal.h" #include "Components/Button.h" #include "Blueprint/UserWidget.h" #include "MyUserWidget.genera…
阅读更多...
微信群发一次能发1000个好友了！

微信群发一次能发1000个好友了！

微信为我们日常交流不可或缺的沟通工具 群发功能在日常生活中使用得也比较多 但有时候 比如在节假日&#xff0c;需要向所有客户发送祝福时 在搞活动时&#xff0c;想向所有客户发起邀约时 如果一个一个点击来发送信息 会非常麻烦 但是&#xff01;&#xff01; 我今天…
阅读更多...
提升Vue3应用效率的秘诀：深入比较ref与reactive！

提升Vue3应用效率的秘诀：深入比较ref与reactive！

ref 和 reactive 是 Vue3 中实现响应式数据的核心 API。ref 用于包装基本数据类型&#xff0c;而 reactive 用于处理对象和数组。尽管 reactive 似乎更适合处理对象&#xff0c;但 Vue3 官方文档更推荐使用 ref。 我的想法&#xff0c;ref就是比reactive好用&#xff0c;官方也…
阅读更多...
安防视频监控平台EasyNVR级联视频上云管理平台EasyNVS，出现报错“i/o deadline reached”该如何解决？

安防视频监控平台EasyNVR级联视频上云管理平台EasyNVS，出现报错“i/o deadline reached”该如何解决？

上云网关管理平台EasyNVS视频综合管理系统具备汇聚与管理EasyGBS、EasyNVR等平台的能力&#xff0c;系统可以将接入的视频资源实现视频能力统一输出&#xff0c;并能进行远程可视化运维等管理功能&#xff0c;还能解决设备现场没有固定公网IP却需要在公网直播的需求。 有用户反…
阅读更多...
最新文章
推荐文章

Copyright @ 2022~2023