更新data文件路径
This commit is contained in:
24
data.yaml
24
data.yaml
@@ -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: 窗户/玻璃门
|
||||
20
datasets/indoor_blind/data.yaml
Normal file
20
datasets/indoor_blind/data.yaml
Normal 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
|
||||
2
train.py
2
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 均可
|
||||
|
||||
Reference in New Issue
Block a user