
ultralytics.utils 子模块解读
本文对 ultralytics.utils 结构下部分子模块进行整体解读。由于该部分代码功能较为分散且文件数量较多,此处汇总部分核心子模块进行学习。
主要模块功能总结
-
autobatch.py - 自动计算最佳批量大小
- 通过分析 GPU 内存使用情况动态确定最优 batch size
- 支持内存使用比例配置
- 包含健壮的错误处理机制
-
autodevice.py - GPU 信息管理和选择
- 查询 NVIDIA GPU 详细统计信息
- 基于利用率和内存选择最空闲的 GPU
- 支持 pynvml 库的健壮错误处理
-
cpu.py - CPU 信息获取
- 跨平台获取 CPU 品牌和型号信息
- 支持 macOS、Linux、Windows


