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 |