From f4cc087b152164969f511103e6591df58b8e9b0e Mon Sep 17 00:00:00 2001
From: tongminqi <1207210763@qq.com>
Date: Sat, 28 Feb 2026 16:05:27 +0800
Subject: [PATCH] Default Changelist

---
 train_segmentation.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train_segmentation.py b/train_segmentation.py
index 99b2cef..0eaadb8 100644
--- a/train_segmentation.py
+++ b/train_segmentation.py
@@ -84,7 +84,7 @@ def main():
     # - 类别: 6个（搭电线、拖车上有车、仪表盘、轮胎、平安马甲、检测仪）
     
     train_segmentation_model(
-        dataset_path="datasets/xianshiping/data.yaml",
+        dataset_path="datasets/xianshiping/data.yaml", # 数据集配置文件路径
         model_name="yolov8n-seg.pt",  # 使用YOLOv8 nano分割模型
         epochs=150,      # 1223张照片，建议150轮（配合早停）
         batch=8,         # CPU/MPS建议8，GPU可以16-32