diff --git a/train_segmentation.py b/train_segmentation.py
index 3df2ed7..99b2cef 100644
--- a/train_segmentation.py
+++ b/train_segmentation.py
@@ -84,7 +84,7 @@ def main():
     # - 类别: 6个（搭电线、拖车上有车、仪表盘、轮胎、平安马甲、检测仪）
     
     train_segmentation_model(
-        dataset_path="datasets/pinganmajia/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