diff --git a/data.yaml b/data.yaml
index 8d18f35..fa16be3 100644
--- a/data.yaml
+++ b/data.yaml
@@ -1,6 +1,6 @@
-train: ../train/images
-val: ../valid/images
-test: ../test/images
+train: datasets/train/images
+val: datasets/valid/images
+test: datasets/test/images
 
 nc: 3
 names: ['button-off', 'button-on', 'gauge']
diff --git a/train.py b/train.py
index c1e44b5..83e930c 100644
--- a/train.py
+++ b/train.py
@@ -48,7 +48,7 @@ TRAIN_CONFIG = {
 def enhanced_data_check():
     """增强型数据验证（稳定版）"""
     print("\n=== 执行增强数据验证 ===")
-    label_files = glob.glob("datasets/insulator/labels/*.txt")
+    label_files = glob.glob("datasets/train/labels/*.txt")
     class_dist = {}
 
     # 类别分布分析
@@ -63,8 +63,8 @@ def enhanced_data_check():
     # 样本可视化（示例保留结构）
     sample_count = 3
     for i in range(sample_count):
-        img_path = f"datasets/insulator/images/{i:04d}.jpg"
-        lbl_path = f"datasets/insulator/labels/{i:04d}.txt"
+        img_path = f"datasets/train/images/{i:04d}.jpg"
+        lbl_path = f"datasets/train/labels/{i:04d}.txt"
         # 添加实际可视化逻辑
 
 def main():