Default Changelist

2026-02-28 16:05:27 +08:00 · 2026-02-28 16:05:27 +08:00 · f4cc087b15
parent 36e5a67e16
commit f4cc087b15
1 changed files with 1 additions and 1 deletions
--- a/train_segmentation.py
+++ b/train_segmentation.py
@ -84,7 +84,7 @@ def main():
    # - 类别: 6个（搭电线、拖车上有车、仪表盘、轮胎、平安马甲、检测仪）
    
    train_segmentation_model(
-        dataset_path="datasets/xianshiping/data.yaml",
+        dataset_path="datasets/xianshiping/data.yaml", # 数据集配置文件路径
        model_name="yolov8n-seg.pt",  # 使用YOLOv8 nano分割模型
        epochs=150,      # 1223张照片，建议150轮（配合早停）
        batch=8,         # CPU/MPS建议8，GPU可以16-32