更新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(
|
results = model.train(
|
||||||
data="data.yaml",
|
data="/home/rongye/ProgramFiles/Yolo/datasets/indoor_blind/data.yaml",
|
||||||
epochs=150, # 导盲任务建议150轮以确保收敛
|
epochs=150, # 导盲任务建议150轮以确保收敛
|
||||||
imgsz=640, # 训练分辨率保持 640
|
imgsz=640, # 训练分辨率保持 640
|
||||||
batch=16, # 3090 显存大,16-32 均可
|
batch=16, # 3090 显存大,16-32 均可
|
||||||
|
|||||||
Reference in New Issue
Block a user