From 66d0877cfdae10eae7aeb7068891c8a9477238e5 Mon Sep 17 00:00:00 2001
From: Asuka <15019597+asuka-civil@user.noreply.gitee.com>
Date: Thu, 29 May 2025 20:16:02 +0800
Subject: [PATCH] =?UTF-8?q?refactor(data):=20=E6=9B=B4=E6=96=B0=E6=95=B0?=
 =?UTF-8?q?=E6=8D=AE=E9=9B=86=E8=B7=AF=E5=BE=84=E5=B9=B6=E8=B0=83=E6=95=B4?=
 =?UTF-8?q?=E6=95=B0=E6=8D=AE=E9=AA=8C=E8=AF=81=E9=80=BB=E8=BE=91?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 修改 data.yaml 中的训练、验证和测试数据集路径
- 更新 train.py 中的数据验证逻辑，使用新的数据集路径
---
 data.yaml | 6 +++---
 train.py  | 6 +++---
 2 files changed, 6 insertions(+), 6 deletions(-)

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():