更新data文件路径

This commit is contained in:
Kevin Wong
2026-01-05 12:19:48 +08:00
parent 4afce79340
commit 0e08372bad
3 changed files with 21 additions and 25 deletions

View File

@@ -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: 窗户/玻璃门

View File

@@ -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

View File

@@ -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 均可