基于深度学习YOLOv8的道路坑洼识别检测系统（YOLOv8+YOLO数据集+UI界面+Python项目源码+模型）

rain: F:\道路坑洼识别检测数据集\train\images val: F:\道路坑洼识别检测数据集\valid\images test: F:\道路坑洼识别检测数据集\test\images nc: 1 names: ['pothole']

from ultralytics import YOLO model_path = 'yolov8s.pt' data_path = 'datasets/data.yaml' if __name__ == '__main__': model = YOLO(model_path) results = model.train(data=data_path, epochs=500, batch=64, device='0', workers=0, project='runs/detect', name='exp', )

fromPyQt5importQtCore, QtGui, QtWidgetsfromPyQt5.QtCoreimportQt, QTimerfromPyQt5.QtGuiimportQImage, QPixmap, QIconfromPyQt5.QtWidgetsimport(QFileDialog, QMessageBox, QTableWidgetItem, QStyledItemDelegate, QHeaderView)importcv2importnumpyasnpfromultralyticsimportYOLOimportosimportdatetimeimportsysclassCenteredDelegate(QStyledItemDelegate):definitStyleOption(self, option, index):super().initStyleOption(option, index) option.displayAlignment = Qt.AlignCenterclassUi_MainWindow(object):defsetupUi(self, MainWindow): MainWindow.setObjectName("MainWindow") MainWindow.resize(1400,900) MainWindow.setWindowTitle("YOLOv8 目标检测系统")# 设置窗口图标ifhasattr(sys,'_MEIPASS'): icon_path = os.path.join(sys._MEIPASS,'icon.ico')else: icon_path ='icon.ico'ifos.path.exists(icon_path): MainWindow.setWindowIcon(QIcon(icon_path)) self.centralwidget = QtWidgets.QWidget(MainWindow) self.centralwidget.setObjectName("centralwidget")# 主布局self.main_layout = QtWidgets.QHBoxLayout(self.centralwidget) self.main_layout.setContentsMargins(10,10,10,10) self.main_layout.setSpacing(15)# 左侧布局 (图像显示)self.left_layout = QtWidgets.QVBoxLayout() self.left_layout.setSpacing(15)# 原始图像组self.original_group = QtWidgets.QGroupBox("原始图像") self.original_group.setMinimumHeight(400) self.original_img_label = QtWidgets.QLabel() self.original_img_label.setAlignment(QtCore.Qt.AlignCenter) self.original_img_label.setText("等待加载图像...") self.original_img_label.setStyleSheet("background-color: #F0F0F0; border: 1px solid #CCCCCC;") original_layout = QtWidgets.QVBoxLayout() original_layout.addWidget(self.original_img_label) self.original_group.setLayout(original_layout) self.left_layout.addWidget(self.original_group)# 检测结果图像组self.result_group = QtWidgets.QGroupBox("检测结果") self.result_group.setMinimumHeight(400) self.result_img_label = QtWidgets.QLabel() self.result_img_label.setAlignment(QtCore.Qt.AlignCenter) self.result_img_label.setText("检测结果将显示在这里") self.result_img_label.setStyleSheet("background-color: #F0F0F0; border: 1px solid #CCCCCC;") result_layout = QtWidgets.QVBoxLayout() result_layout.addWidget(self.result_img_label) self.result_group.setLayout(result_layout) self.left_layout.addWidget(self.result_group) self.main_layout.addLayout(self.left_layout, stretch=3)# 右侧布局 (控制面板)self.right_layout = QtWidgets.QVBoxLayout() self.right_layout.setSpacing(15)# 模型选择组self.model_group = QtWidgets.QGroupBox("模型设置") self.model_group.setStyleSheet("QGroupBox { font-weight: bold; }") self.model_layout = QtWidgets.QVBoxLayout()# 模型选择self.model_combo = QtWidgets.QComboBox() self.model_combo.addItems(["best.pt"]) self.model_combo.setCurrentIndex(0)# 加载模型按钮self.load_model_btn = QtWidgets.QPushButton(" 加载模型") self.load_model_btn.setIcon(QIcon.fromTheme("document-open")) self.load_model_btn.setStyleSheet("QPushButton { padding: 8px; background-color: #4CAF50; color: white; border-radius: 4px; }""QPushButton:hover { background-color: #45a049; }") self.model_layout.addWidget(self.model_combo) self.model_layout.addWidget(self.load_model_btn) self.model_group.setLayout(self.model_layout) self.right_layout.addWidget(self.model_group)# 参数设置组self.param_group = QtWidgets.QGroupBox("检测参数") self.param_group.setStyleSheet("QGroupBox { font-weight: bold; }") self.param_layout = QtWidgets.QFormLayout() self.param_layout.setLabelAlignment(Qt.AlignLeft) self.param_layout.setFormAlignment(Qt.AlignLeft) self.param_layout.setVerticalSpacing(15)# 置信度滑块self.conf_slider = QtWidgets.QSlider(Qt.Horizontal) self.conf_slider.setRange(1,99) self.conf_slider.setValue(25) self.conf_value = QtWidgets.QLabel("0.25") self.conf_value.setAlignment(Qt.AlignCenter) self.conf_value.setStyleSheet("font-weight: bold; color: #2196F3;")# IoU滑块self.iou_slider = QtWidgets.QSlider(Qt.Horizontal) self.iou_slider.setRange(1,99) self.iou_slider.setValue(45) self.iou_value = QtWidgets.QLabel("0.45") self.iou_value.setAlignment(Qt.AlignCenter) self.iou_value.setStyleSheet("font-weight: bold; color: #2196F3;") self.param_layout.addRow("置信度阈值:", self.conf_slider) self.param_layout.addRow("当前值:", self.conf_value) self.param_layout.addRow(QtWidgets.QLabel(""))# 空行self.param_layout.addRow("IoU阈值:", self.iou_slider) self.param_layout.addRow("当前值:", self.iou_value) self.param_group.setLayout(self.param_layout) self.right_layout.addWidget(self.param_group)# 功能按钮组self.func_group = QtWidgets.QGroupBox("检测功能") self.func_group.setStyleSheet("QGroupBox { font-weight: bold; }") self.func_layout = QtWidgets.QVBoxLayout() self.func_layout.setSpacing(10)# 图片检测按钮self.image_btn = QtWidgets.QPushButton(" 图片检测") self.image_btn.setIcon(QIcon.fromTheme("image-x-generic"))# 视频检测按钮self.video_btn = QtWidgets.QPushButton(" 视频检测") self.video_btn.setIcon(QIcon.fromTheme("video-x-generic"))# 摄像头检测按钮self.camera_btn = QtWidgets.QPushButton(" 摄像头检测") self.camera_btn.setIcon(QIcon.fromTheme("camera-web"))# 停止检测按钮self.stop_btn = QtWidgets.QPushButton(" 停止检测") self.stop_btn.setIcon(QIcon.fromTheme("process-stop")) self.stop_btn.setEnabled(False)# 保存结果按钮self.save_btn = QtWidgets.QPushButton(" 保存结果") self.save_btn.setIcon(QIcon.fromTheme("document-save")) self.save_btn.setEnabled(False)# 设置按钮样式button_style =""" QPushButton { padding: 10px; background-color: #2196F3; color: white; border: none; border-radius: 4px; text-align: left; } QPushButton:hover { background-color: #0b7dda; } QPushButton:disabled { background-color: #cccccc; } """forbtnin[self.image_btn, self.video_btn, self.camera_btn, self.stop_btn, self.save_btn]: btn.setStyleSheet(button_style) self.func_layout.addWidget(btn) self.func_group.setLayout(self.func_layout) self.right_layout.addWidget(self.func_group)# 检测结果表格组self.table_group = QtWidgets.QGroupBox("检测结果详情") self.table_group.setStyleSheet("QGroupBox { font-weight: bold; }") self.table_layout = QtWidgets.QVBoxLayout() self.result_table = QtWidgets.QTableWidget() self.result_table.setColumnCount(4) self.result_table.setHorizontalHeaderLabels(["类别","置信度","左上坐标","右下坐标"]) self.result_table.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch) self.result_table.verticalHeader().setVisible(False) self.result_table.setSelectionBehavior(QtWidgets.QAbstractItemView.SelectRows) self.result_table.setEditTriggers(QtWidgets.QAbstractItemView.NoEditTriggers)# 设置表格样式self.result_table.setStyleSheet(""" QTableWidget { border: 1px solid #e0e0e0; alternate-background-color: #f5f5f5; } QHeaderView::section { background-color: #2196F3; color: white; padding: 5px; border: none; } QTableWidget::item { padding: 5px; } """)# 设置居中代理delegate = CenteredDelegate(self.result_table) self.result_table.setItemDelegate(delegate) self.table_layout.addWidget(self.result_table) self.table_group.setLayout(self.table_layout) self.right_layout.addWidget(self.table_group, stretch=1) self.main_layout.addLayout(self.right_layout, stretch=1) MainWindow.setCentralWidget(self.centralwidget)# 状态栏self.statusbar = QtWidgets.QStatusBar(MainWindow) self.statusbar.setStyleSheet("QStatusBar { border-top: 1px solid #c0c0c0; }") MainWindow.setStatusBar(self.statusbar)# 初始化变量self.model =Noneself.cap =Noneself.timer = QTimer() self.is_camera_running =Falseself.current_image =Noneself.current_result =Noneself.video_writer =Noneself.output_path ="output"# 创建输出目录ifnotos.path.exists(self.output_path): os.makedirs(self.output_path)# 连接信号槽self.load_model_btn.clicked.connect(self.load_model) self.image_btn.clicked.connect(self.detect_image) self.video_btn.clicked.connect(self.detect_video) self.camera_btn.clicked.connect(self.detect_camera) self.stop_btn.clicked.connect(self.stop_detection) self.save_btn.clicked.connect(self.save_result) self.conf_slider.valueChanged.connect(self.update_conf_value) self.iou_slider.valueChanged.connect(self.update_iou_value) self.timer.timeout.connect(self.update_camera_frame)# 设置全局样式self.set_style()defset_style(self): style =""" QMainWindow { background-color: #f5f5f5; } QGroupBox { border: 1px solid #e0e0e0; border-radius: 5px; margin-top: 10px; padding-top: 15px; } QGroupBox::title { subcontrol-origin: margin; left: 10px; padding: 0 3px; } QLabel { color: #333333; } QComboBox { padding: 5px; border: 1px solid #cccccc; border-radius: 3px; } QSlider::groove:horizontal { height: 6px; background: #e0e0e0; border-radius: 3px; } QSlider::handle:horizontal { width: 16px; height: 16px; margin: -5px 0; background: #2196F3; border-radius: 8px; } QSlider::sub-page:horizontal { background: #2196F3; border-radius: 3px; } """self.centralwidget.setStyleSheet(style)defload_model(self): model_name = self.model_combo.currentText().split(" ")[0]try: self.model = YOLO(model_name) self.statusbar.showMessage(f"模型{model_name}加载成功",3000) self.image_btn.setEnabled(True) self.video_btn.setEnabled(True) self.camera_btn.setEnabled(True)exceptExceptionase: QMessageBox.critical(None,"错误",f"模型加载失败:{str(e)}")defupdate_conf_value(self): conf = self.conf_slider.value() /100self.conf_value.setText(f"{conf:.2f}")defupdate_iou_value(self): iou = self.iou_slider.value() /100self.iou_value.setText(f"{iou:.2f}")defdetect_image(self):ifself.modelisNone: QMessageBox.warning(None,"警告","请先加载模型")returnfile_path, _ = QFileDialog.getOpenFileName(None,"选择图片","","图片文件 (*.jpg *.jpeg *.png *.bmp);;所有文件 (*)")iffile_path:try:# 读取图片img = cv2.imread(file_path) img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)# 显示原始图片self.display_image(img, self.original_img_label) self.current_image = img.copy()# 检测图片conf = self.conf_slider.value() /100iou = self.iou_slider.value() /100self.statusbar.showMessage("正在检测图片...") QtWidgets.QApplication.processEvents()# 更新UIresults = self.model.predict(img, conf=conf, iou=iou) result_img = results[0].plot()# 显示检测结果self.display_image(result_img, self.result_img_label) self.current_result = result_img.copy()# 更新结果表格self.update_result_table(results[0]) self.save_btn.setEnabled(True) self.statusbar.showMessage(f"图片检测完成:{os.path.basename(file_path)}",3000)exceptExceptionase: QMessageBox.critical(None,"错误",f"图片检测失败:{str(e)}") self.statusbar.showMessage("图片检测失败",3000)defdetect_video(self):ifself.modelisNone: QMessageBox.warning(None,"警告","请先加载模型")returnfile_path, _ = QFileDialog.getOpenFileName(None,"选择视频","","视频文件 (*.mp4 *.avi *.mov *.mkv);;所有文件 (*)")iffile_path:try: self.cap = cv2.VideoCapture(file_path)ifnotself.cap.isOpened():raiseException("无法打开视频文件")# 获取视频信息fps = self.cap.get(cv2.CAP_PROP_FPS) width =int(self.cap.get(cv2.CAP_PROP_FRAME_WIDTH)) height =int(self.cap.get(cv2.CAP_PROP_FRAME_HEIGHT))# 创建视频写入器timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S") output_file = os.path.join(self.output_path,f"output_{timestamp}.mp4") fourcc = cv2.VideoWriter_fourcc(*'mp4v') self.video_writer = cv2.VideoWriter(output_file, fourcc, fps, (width, height))# 启用停止按钮，禁用其他按钮self.stop_btn.setEnabled(True) self.save_btn.setEnabled(True) self.image_btn.setEnabled(False) self.video_btn.setEnabled(False) self.camera_btn.setEnabled(False)# 开始处理视频self.timer.start(30)# 30ms间隔self.statusbar.showMessage(f"正在处理视频:{os.path.basename(file_path)}...")exceptExceptionase: QMessageBox.critical(None,"错误",f"视频检测失败:{str(e)}") self.statusbar.showMessage("视频检测失败",3000)