# ===================================================================
# 自动化训练与导出配置文件
# ===================================================================

# --- 项目与模型设置 ---
# 项目名称，所有训练结果将保存在 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