From 0e08372badc6ff4244d4424dee42338abfa9aaeb Mon Sep 17 00:00:00 2001
From: Kevin Wong <lamnickdavid@gmail.com>
Date: Mon, 5 Jan 2026 12:19:48 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0data=E6=96=87=E4=BB=B6?=
 =?UTF-8?q?=E8=B7=AF=E5=BE=84?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 data.yaml                       | 24 ------------------------
 datasets/indoor_blind/data.yaml | 20 ++++++++++++++++++++
 train.py                        |  2 +-
 3 files changed, 21 insertions(+), 25 deletions(-)
 delete mode 100644 data.yaml
 create mode 100644 datasets/indoor_blind/data.yaml

diff --git a/data.yaml b/data.yaml
deleted file mode 100644
index 7d9ee0c..0000000
--- a/data.yaml
+++ /dev/null
@@ -1,24 +0,0 @@
-# 导盲分割数据集 - 从 MIT Indoor 筛选的 14 个核心类别
-# 原始 2573 类别 -> 14 个导盲相关类别
-
-path: /home/rongye/ProgramFiles/Yolo/datasets/indoor_blind
-train: train/images
-val: valid/images
-test: test/images
-
-nc: 14
-names:
-  - floor       # 0: 可行走地面
-  - corridor    # 1: 走廊/通道
-  - sidewalk    # 2: 人行道
-  - chair       # 3: 椅子
-  - table       # 4: 桌子
-  - sofa_bed    # 5: 沙发/床
-  - door        # 6: 门
-  - elevator    # 7: 电梯
-  - stairs      # 8: 楼梯
-  - wall        # 9: 墙壁
-  - person      # 10: 行人
-  - cabinet     # 11: 柜子
-  - trash_can   # 12: 垃圾桶
-  - window      # 13: 窗户/玻璃门
\ No newline at end of file
diff --git a/datasets/indoor_blind/data.yaml b/datasets/indoor_blind/data.yaml
new file mode 100644
index 0000000..318e6d5
--- /dev/null
+++ b/datasets/indoor_blind/data.yaml
@@ -0,0 +1,20 @@
+names:
+- floor
+- corridor
+- sidewalk
+- chair
+- table
+- sofa_bed
+- door
+- elevator
+- stairs
+- wall
+- person
+- cabinet
+- trash_can
+- window
+nc: 14
+path: /home/rongye/ProgramFiles/Yolo/datasets/indoor_blind
+test: test/images
+train: train/images
+val: valid/images
diff --git a/train.py b/train.py
index 30d8d4e..6c24c73 100644
--- a/train.py
+++ b/train.py
@@ -6,7 +6,7 @@ model = YOLO("/home/rongye/ProgramFiles/Yolo/yolo11l-seg.pt")
 
 # 开始训练
 results = model.train(
-    data="data.yaml",
+    data="/home/rongye/ProgramFiles/Yolo/datasets/indoor_blind/data.yaml",
     epochs=150,           # 导盲任务建议150轮以确保收敛
     imgsz=640,            # 训练分辨率保持 640
     batch=16,             # 3090 显存大，16-32 均可