Ultralytics 框架中的 model.tune()
方法 是一种基于遗传算法(Genetic Algorithm, GA)的超参数自动优化工具,旨在通过智能搜索找到最佳超参数组合,从而提升模型性能。以下从作用、参数详解、使用场景及最佳实践进行解析:
一、model.tune()
的核心作用
-
自动化超参数优化
通过遗传算法在指定搜索空间中探索超参数组合,无需手动反复试验,显著提升调参效率。 -
提升模型性能
优化学习率、批次大小、数据增强强度等关键参数,最大化模型的精度(如 mAP)或推理速度。 -
适配多样场景
支持目标检测、实例分割、姿态估计等多种任务,适配不同硬件环境(CPU/GPU/边缘设备)。
二、核心参数详解
以下为 model.tune()
的核心参数及其作用:
参数 | 类型 | 默认值 | |
---|