在 YOLOv5 的源代码中,utils
文件夹包含了许多实用的工具函数和类。这些函数和类用于完成各种任务,例如数据加载、图像处理、模型评估等。例如,datasets.py
文件中定义了用于加载数据的类,general.py
文件中定义了一些通用的实用函数,torch_utils.py
文件中定义了一些与 PyTorch 相关的实用函数等。里面有loss函数,metrics函数,plots函数等等。
aws # 恢复中断训练,和aws平台使用相关的工具
flask_rest_api # 和flask 相关的工具
google_app_engine # 和谷歌app引擎相关的工具
loggers # 日志打打印
_init_.py # notebook的初始化,检查系统软件和硬件
activations.py # 激活函数
augmentations # 存放各种图像增强技术
autoanchor.py # 自动生成锚框
autobatch.py # 自动生成批量大小
benchmarks.py # 对模型进行性能评估(推理速度和内存占用上的评估)
callbacks.py # 回调函数,主要为logger服务
datasets # dateset和dateloader定义代码
downloads.py # 谷歌云盘内容下载
general.py # 全项目通用代码,相关实用函数实现
loss.py # 存放各种损失函数
metrics.py # 模型验证指标,包括ap,混淆矩阵等
plots.py # 绘图相关函数,如绘制loss、ac曲线,还能单独将一个bbox存储为图像
torch_utils.py # 辅助函数