基于YOLOv8道路汽车漆面缺陷（划痕）检测系统

文章目录

道路汽车漆面车漆缺陷检测数据集1221张 1类
在这里插入图片描述

汽车漆面缺陷检测YOLO数据集 1221张，
按照8比2划分为训练集和验证集，其中训练集976张，验证集245张，模型分为1类，
在这里插入图片描述

分类为：【‘scratch’】
图片数量和标注框数量：
scratch: 图片数【1221】，标注框数【1965】
在这里插入图片描述
基于YOLOv8的汽车漆面缺陷（划痕）检测系统，我们将从安装依赖、准备数据集、数据预处理、配置YOLOv8、训练和评估模型，到构建GUI应用程序进行详细的介绍。以下是具体步骤和代码示例。

1. 安装依赖

首先确保你已经安装了必要的Python包：

pip install ultralytics opencv-python-headless matplotlib PySide6

ultralytics包含了YOLOv8的所有实现，opencv-python-headless用于图像处理，matplotlib可以用来可视化结果，而PySide6将用于创建图形用户界面（GUI）。
在这里插入图片描述

2. 数据集准备与划分

假设同学你的，数据集已经按照YOLO格式准备好，并且需要按照8:2的比例划分为训练集和验证集。同学呀你需要创建一个data.yaml文件来描述数据集路径和类别信息：

train: /path/to/train/images/
val: /path/to/val/images/

nc: 1 # 类别数量
names: ['scratch']

请替换/path/to/train/images/和/path/to/val/images/为同学的实际数据集路径。

3. 数据预处理

使用的是YOLO格式的数据集，大部分预处理工作（如调整图片大小、归一化等）将由YOLOv8自动完成。因此，此步骤主要是指确保数据集正确地组织并符合要求。

4. 配置YOLOv8

加载YOLOv8n模型，开始配置：

from ultralytics import YOLO

# 加载YOLOv8n模型
model = YOLO('yolov8n.yaml')

5. 训练和评估模型

加载数据集并开始训练：

# 开始训练
results = model.train(
    data='/path/to/data.yaml', 
    epochs=100, 
    imgsz=640, 
    batch=16,  # 根据你的GPU内存大小调整
    patience=50,  # 提前停止的耐心值
    lr0=0.01,  # 初始学习率
    lrf=0.1,  # 最终学习率
    optimizer='SGD',  # 可选：'Adam', 'RMSProp'
    device='0',  # 使用第0号GPU，'-1'表示使用CPU
    workers=8,  # 数据加载器的工作线程数
    verbose=True
)

# 模型评估
metrics = model.val()
print(f"Validation mAP: {metrics.box.map}")  # 打印mAP值以评估模型性能

6. 推理与可视化

完成训练后，您可以使用模型进行推理，并可视化结果：

def predict_image(model, img_path):
    results = model.predict(source=img_path, save=True)
    for r in results:
        im_array = r.plot()  # 绘制预测结果
        im = Image.fromarray(im_array[..., ::-1])  # 转换颜色通道顺序
        return im

img_path = '/path/to/test/image.jpg'
im = predict_image(model, img_path)
im.show()  # 显示图片

7. 构建GUI应用程序

那么呢？怎搞？接下来，我们使用PySide6构建一个简单的GUI应用程序，用于选择图片并显示预测结果：

import sys
from PySide6.QtWidgets import QApplication, QMainWindow, QPushButton, QVBoxLayout, QWidget, QLabel, QFileDialog
from PIL import Image
from ultralytics import YOLO

class ScratchDetectionApp(QMainWindow):
    def __init__(self):
        super().__init__()
        self.setWindowTitle("汽车漆面缺陷检测")
        self.setGeometry(100, 100, 800, 600)

        self.model = YOLO('yolov8n.yaml')
        
        self.initUI()

    def initUI(self):
        self.layout = QVBoxLayout()

        self.image_label = QLabel(self)
        self.image_label.setAlignment(Qt.AlignCenter)
        self.layout.addWidget(self.image_label)

        self.load_button = QPushButton("加载图片", self)
        self.load_button.clicked.connect(self.load_image)
        self.layout.addWidget(self.load_button)

        container = QWidget()
        container.setLayout(self.layout)
        self.setCentralWidget(container)

    def load_image(self):
        file_dialog = QFileDialog()
        file_dialog.setNameFilter("Images (*.png *.jpg *.jpeg)")
        if file_dialog.exec():
            file_name = file_dialog.selectedFiles()[0]
            self.predict_and_display(file_name)

    def predict_and_display(self, file_name):
        im = predict_image(self.model, file_name)
        pixmap = QPixmap.fromImage(ImageQt.ImageQt(im))
        self.image_label.setPixmap(pixmap)

if __name__ == "__main__":
    app = QApplication(sys.argv)
    window = ScratchDetectionApp()
    window.show()
    sys.exit(app.exec())