首页分享如何使用yolov8训练使用——马铃薯病害检测数据集，共8000张，voc或yolo格式标注土豆数据集病害检测

如何使用yolov8训练使用——马铃薯病害检测数据集，共8000张，voc或yolo格式标注土豆数据集病害检测

来源：花匠小妙招时间：2024-12-22 01:19

马铃薯病害检测数据集，包含受损马铃薯、缺陷马铃薯、病菌马铃薯、正常马铃薯、发芽马铃薯，共8000张，voc或yolo格式标注
在这里插入图片描述
马铃薯病害检测数据集
数据集描述
该数据集包含8000张马铃薯的图片及其对应的标注，标注格式为VOC或YOLO。数据集分为五类：受损马铃薯、缺陷马铃薯、病菌马铃薯、正常马铃薯、发芽马铃薯。每个标注文件包含了每张图片中马铃薯的位置信息。

数据集结构
假设数据集的目录结构如下：

深色版本
potato_disease_dataset/
├── images/
│ ├── train/
│ ├── valid/
│ ├── test/
├── labels/
│ ├── train/
│ ├── valid/
│ ├── test/
数据集统计
图片总数：8000张
标注总数：根据实际标注情况确定
类别数：5
类别名称：
损坏马铃薯 (damaged_potato)
缺陷马铃薯 (defective_potato)
病菌马铃薯 (diseased_potato)
正常马铃薯 (healthy_potato)
发芽马铃薯 (sprouted_potato)
配置文件
创建一个配置文件 potato_disease.yaml，内容如下：

yaml
深色版本
path: potato_disease_dataset # 数据集路径
train: images/train # 训练集图像路径
val: images/valid # 验证集图像路径
test: images/test # 测试集图像路径

nc: 5 # 类别数
names: [‘damaged_potato’, ‘defective_potato’, ‘diseased_potato’, ‘healthy_potato’, ‘sprouted_potato’] # 类别名称
训练模型
使用YOLOv8进行训练的代码如下：

环境准备
确保已安装以下软件包：

Python
PyTorch
Ultralytics
你可以使用pip安装它们：

bash
深色版本
pip install torch torchvision
pip install ultralytics
训练模型
python
深色版本
from ultralytics import YOLO

加载预训练模型

model = YOLO(‘yolov8n.pt’)

训练模型

results = model.train(
data=‘potato_disease.yaml’, # 数据集配置文件
epochs=100, # 训练轮数
imgsz=640, # 图像尺寸
batch=16, # 批量大小
name=‘potato_disease_detection’ # 实验名称
)
模型评估
训练完成后，可以使用以下命令评估模型在验证集上的性能：

python
深色版本

评估模型

results = model.val(data=‘potato_disease.yaml’, imgsz=640)
可视化结果
你可以使用以下代码来可视化模型的预测结果：

python
深色版本
import cv2
import torch
from ultralytics import YOLO

加载训练好的模型

model = YOLO(‘runs/detect/potato_disease_detection/weights/best.pt’)

读取测试图像

image_path = ‘potato_disease_dataset/images/test/image1.jpg’
image = cv2.imread(image_path)

进行预测

results = model(image)

可视化预测结果

for result in results:
boxes = result.boxes.xyxy # 获取边界框
confidences = result.boxes.conf # 获取置信度
class_ids = result.boxes.cls # 获取类别ID

for box, confidence, class_id in zip(boxes, confidences, class_ids): x1, y1, x2, y2 = map(int, box) label = f'{["damaged_potato", "defective_potato", "diseased_potato", "healthy_potato", "sprouted_potato"][int(class_id)]} {confidence:.2f}' color = (0, 255, 0) # 绿色 cv2.rectangle(image, (x1, y1), (x2, y2), color, 2) cv2.putText(image, label, (x1, y1 - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.5, color, 2) 123456

显示图像

cv2.imshow(‘Prediction’, image)
cv2.waitKey(0)
cv2.destroyAllWindows()
改进策略
为了提高模型的性能，可以尝试以下改进策略：

数据增强：
使用更多的数据增强技术，如随机裁剪、旋转、翻转等。
使用MixUp、CutMix等高级数据增强方法。
模型架构改进：
替换主干网络，例如使用ResNet、EfficientNet等更强大的主干网络。
引入注意力机制，如SENet、CBAM等。
使用更高效的特征融合方法，如BiFPN、ASFF等。
优化器和学习率调度：
尝试不同的优化器，如AdamW、RMSprop等。
使用更复杂的学习率调度策略，如Cosine Annealing、OneCyclePolicy等。
损失函数：
尝试不同的损失函数，如Focal Loss、Dice Loss等。
正则化：
使用Dropout、L2正则化等技术防止过拟合。
多尺度训练：
在不同尺度的图像上进行训练，提高模型的泛化能力。
集成学习：
训练多个模型，然后进行集成，提高整体性能。
示例代码
以下是一个示例代码，展示如何在训练时应用一些常见的改进策略：

python
深色版本
from ultralytics import YOLO
import albumentations as A
from albumentations.pytorch import ToTensorV2

定义数据增强

transform = A.Compose([
A.RandomCrop(width=450, height=450),
A.HorizontalFlip(p=0.5),
A.VerticalFlip(p=0.5),
A.RandomRotate90(p=0.5),
A.Cutout(num_holes=8, max_h_size=64, max_w_size=64, fill_value=0, p=0.5),
A.Normalize(mean=(0.485, 0.456, 0.406), std=(0.229, 0.224, 0.225)),
ToTensorV2()
])

加载预训练模型

model = YOLO(‘yolov8n.pt’)

训练模型

results = model.train(
data=‘potato_disease.yaml’, # 数据集配置文件
epochs=100, # 训练轮数
imgsz=640, # 图像尺寸
batch=16, # 批量大小
name=‘potato_disease_detection_improved’, # 实验名称
augment=True, # 使用数据增强
transform=transform # 自定义数据增强
)
总结
以上步骤涵盖了从数据集准备到模型训练、评估和可视化的完整流程，并提供了一些常见的改进策略。