当前位置：首页 > news >正文

yolov8-seg裂缝分割模型包含裂缝分割数据集 4000张

news 2026/6/22 23:09:23

yolov8-seg裂缝分割模型包含裂缝分割数据集 4000张
训练好的yol ov8-seg裂缝分割模型包含裂缝分割数据集4000张(配置环境后，项目可自己跑)
[1们]训练了yolov8n-seg，yolov8m- seg，yolov8x-seg三个模型，训练了100epoch
[2]训练任务为分割
segment
[3]训练数据集和检测结果如图1，4000张yol o格式的裂缝分割数据
文件包含:
[1] yolo源码，
[2]裂缝分割数据集(4000张)
[3]训练好的权重(yolov8n-seg, yolov8m-seg， yolov8x-seg)

这个裂缝分割项目，并提供必要的脚本和配置文件。这个项目包含4000张裂缝分割数据集，已经训练了YOLOv8-seg的三个不同规模的模型（yolov8n-seg、yolov8m-seg、yolov8x-seg），每个模型训练了100个epoch。我们将提供如何配置环境、训练模型和进行预测的详细步骤。

1. 项目结构

假设你的项目目录结构如下：

crack_segmentation_project/ ├── dataset/ │ ├── images/ │ │ ├── train/ │ │ └── val/ │ ├── labels/ │ │ ├── train/ │ │ └── val/ ├── models/ │ ├── yolov8n-seg.pt │ ├── yolov8m-seg.pt │ ├── yolov8x-seg.pt ├── yolo/ │ └── (YOLOv8源码) ├── data.yaml ├── train.py ├── predict.py └── requirements.txt

2. 数据集配置文件 (`data.yaml`)

创建一个data.yaml文件，配置数据集的路径和类别信息：

path:./dataset# 数据集路径train:images/train# 训练集图像路径val:images/val# 验证集图像路径nc:1# 类别数（裂缝）names:-crack# 类别名称

3. 安装依赖

创建一个requirements.txt文件，列出所需的依赖项：

torch ultralytics opencv-python

安装依赖项：

pipinstall-rrequirements.txt

4. 训练脚本 (`train.py`)

fromultralyticsimportYOLOdeftrain_model(data_yaml_path,model_config,epochs,batch_size,img_size,augment):# 加载模型model=YOLO(model_config)# 训练模型results=model.train(data=data_yaml_path,epochs=epochs,batch=batch_size,imgsz=img_size,augment=augment)# 保存模型model.save(f"models/{model_config.split('/')[-1].replace('.yaml','')}.pt")if__name__=="__main__":data_yaml_path='data.yaml'model_configs=['yolo/models/yolov8n-seg.yaml','yolo/models/yolov8m-seg.yaml','yolo/models/yolov8x-seg.yaml']epochs=100batch_size=16img_size=640augment=Trueformodel_configinmodel_configs:train_model(data_yaml_path,model_config,epochs,batch_size,img_size,augment)

5. 预测脚本 (`predict.py`)

importcv2importtorchfromultralyticsimportYOLOdefpredict_image(image_path,model_path,img_size=640):# 加载模型model=YOLO(model_path)# 读取图像image=cv2.imread(image_path)image_rgb=cv2.cvtColor(image,cv2.COLOR_BGR2RGB)# 进行预测results=model(image_rgb,size=img_size)# 处理预测结果forresultinresults:masks=result.masks.data.cpu().numpy()formaskinmasks:mask=(mask*255).astype('uint8')colored_mask=cv2.applyColorMap(mask,cv2.COLORMAP_JET)image=cv2.addWeighted(image,1,colored_mask,0.5,0)# 显示图像cv2.imshow('Prediction',image)cv2.waitKey(0)cv2.destroyAllWindows()if__name__=="__main__":image_path='path_to_your_image.jpg'model_path='models/yolov8n-seg.pt'# 选择你要使用的模型predict_image(image_path,model_path)

6. 运行脚本

训练模型：
```
python train.py
```
进行预测：
```
python predict.py
```

7. 详细解释

数据集配置文件 (`data.yaml`)

path: 数据集的根目录路径。
train: 训练集图像的路径。
val: 验证集图像的路径。
nc: 类别数（裂缝）。
names: 类别名称列表。

训练脚本 (`train.py`)

导入依赖项：
- from ultralytics import YOLO：导入YOLOv8模型。
定义训练函数：
- train_model：加载模型，设置训练参数，训练模型，并保存最佳模型。
主函数：
- 设置数据集路径、模型配置、训练参数等。
- 遍历不同的模型配置文件，调用train_model函数进行训练。

预测脚本 (`predict.py`)

导入依赖项：
- import cv2：导入OpenCV库。
- import torch：导入PyTorch库。
- from ultralytics import YOLO：导入YOLOv8模型。
定义预测函数：
- predict_image：加载模型，读取图像，进行预测，处理预测结果，并显示带有分割结果的图像。
主函数：
- 设置图像路径和模型路径。
- 调用predict_image函数进行预测。