32 lines
1.0 KiB
YAML
32 lines
1.0 KiB
YAML
|
# ===================================================================
|
|||
|
|
# 自动化训练与导出配置文件
|
|||
|
|
# ===================================================================
|
|||
|
|
|
|||
|
|
# --- 项目与模型设置 ---
|
|||
|
|
# 项目名称，所有训练结果将保存在 runs/detect/{project_name} 文件夹下
|
|||
|
|
project_name: 'Safety_Detection_Project'
|
|||
|
|
|
|||
|
|
# 基础模型：可以是官方的预训练模型 (如 yolov8n.pt), 也可以是你自己的 .pt 文件路径
|
|||
|
|
base_model: 'yolo11n.pt'
|
|||
|
|
|
|||
|
|
# 数据集配置文件：【非常重要】请务必修改为你自己的 data .yaml 文件的绝对或相对路径
|
|||
|
|
data_yaml: 'coco128.yaml'
|
|||
|
|
|
|||
|
|
# --- 流程控制 ---
|
|||
|
|
# 是否执行训练步骤
|
|||
|
|
run_training: false
|
|||
|
|
# 是否执行导出步骤
|
|||
|
|
run_export: true
|
|||
|
|
|
|||
|
|
# --- 训练参数 ---
|
|||
|
|
epochs: 100
|
|||
|
|
imgsz: 640
|
|||
|
|
batch_size: 16
|
|||
|
|
# 使用多少个CPU核心进行数据加载，0表示只用主进程
|
|||
|
|
workers: 8
|
|||
|
|
|
|||
|
|
# --- 导出参数 ---
|
|||
|
|
# 目标格式 (e.g., onnx, tensorrt, coreml)
|
|||
|
|
export_format: 'onnx'
|
|||
|
|
# 是否使用半精度(FP16)导出，可以提速并减小文件大小
|
|||
|
|
half_precision: true
|