基于 PyTorch 的类激活图(CAM)方法库,用于可视化视觉模型做出预测时关注的图像区域。
ultralytics/yolov5
基于 PyTorch 的计算机视觉工具库,支持目标检测、实例分割和图像分类的训练与模型导出。
- Star
- 58,185
- Fork
- 17,457
- 未关闭 Issue
- 25
- 最近提交
- 前天
这是什么?
YOLOv5 是 Ultralytics 基于 PyTorch 开发的计算机视觉模型库,覆盖目标检测、实例分割和图像分类三类任务。仓库提供在自定义数据集上训练、精度验证,以及对图片、视频、摄像头和视频流进行推理的脚本,训练好的权重还可以导出。
项目提供从 nano 到 extra-large 多个尺寸的预训练模型,分别基于 COCO 和 ImageNet 训练。用户可以通过 PyTorch Hub 直接加载模型,也可以使用 detect.py 命令行运行推理。训练支持多 GPU 分布式数据并行,导出工具可将模型转换为 ONNX、TensorRT、CoreML 和 TFLite 格式。
文档还包含迁移学习、超参数进化、模型剪枝、测试时增强,以及与 ClearML、Comet、Weights & Biases 等工具的实验记录集成教程。项目同时说明,目前持续维护的版本是 ultralytics pip 包,它提供更新的 YOLO 模型,并支持姿态估计、旋转框检测等任务。
核心功能
- 01在同一代码库中完成目标检测、实例分割和图像分类。
- 02提供从 nano 到 extra-large 的预训练模型,可通过 PyTorch Hub 或 detect.py 调用。
- 03支持图片、视频、摄像头、屏幕截图、RTSP 流和 YouTube 链接的推理。
- 04支持多 GPU 分布式训练,并可通过 AutoBatch 自动选择批大小。
- 05支持导出为 ONNX、TensorRT、CoreML 和 TFLite 格式。
适用场景
- 在自定义标注数据集上训练目标检测模型。
- 使用摄像头或视频文件进行实时目标检测。
- 通过 TensorRT 或 TFLite 将训练好的模型部署到边缘设备。
- 借助预训练权重快速试验分割或分类任务。
快速上手
- 1
克隆仓库
git clone https://github.com/ultralytics/yolov5 - 2
进入克隆的目录
cd yolov5 - 3
安装所需依赖
pip install -r requirements.txt - 4
对本地图片运行推理
python detect.py --weights yolov5s.pt --source img.jpg
使用前须知
- 采用 AGPL-3.0 许可证,属于 Copyleft 许可,衍生的网络服务可能需要开源。
- 若要在不受 AGPL 约束的情况下进行企业使用,需向 Ultralytics 单独申请授权。
- README 指出,若需要更新的架构,应使用持续维护的 ultralytics 包。
标签
常见问题
- yolov5 是什么?
- 基于 PyTorch 的计算机视觉工具库,支持目标检测、实例分割和图像分类的训练与模型导出。 YOLOv5 是 Ultralytics 基于 PyTorch 开发的计算机视觉模型库,覆盖目标检测、实例分割和图像分类三类任务。仓库提供在自定义数据集上训练、精度验证,以及对图片、视频、摄像头和视频流进行推理的脚本,训练好的权重还可以导出。
- yolov5 可以免费使用吗?用的什么许可证?
- yolov5 以 AGPL-3.0 许可证开源。用于商业产品前,请先确认许可证条款。
- yolov5 还在维护吗?
- 活跃维护中。yolov5 最近一次提交是在 2026年10月18日。
- yolov5 适合用来做什么?
- 在自定义标注数据集上训练目标检测模型。 使用摄像头或视频文件进行实时目标检测。 通过 TensorRT 或 TFLite 将训练好的模型部署到边缘设备。 借助预训练权重快速试验分割或分类任务。
- yolov5 怎么安装和运行?
- 克隆仓库: git clone https://github.com/ultralytics/yolov5 进入克隆的目录: cd yolov5 安装所需依赖: pip install -r requirements.txt 对本地图片运行推理: python detect.py --weights yolov5s.pt --source img.jpg
- 在哪里获取 yolov5?
- 源代码和安装说明在 GitHub 上:https://github.com/ultralytics/yolov5
概要由 AI 根据项目 README 生成;GitHub 数据每日更新。
相关项目
一款跨平台桌面应用,用 AI 辅助标注文本、图像、视频、点云和多模态数据,支持多种格式导出。
BiRefNet 官方 PyTorch 实现,用于高分辨率二分图像分割,常用于抠图去背景。
收录 419 个预训练 AI 模型,覆盖视觉、语音、NLP 与大模型,可通过统一的 Python 命令行运行。
面向大模型分布式训练与推理的深度学习优化库,提升效率并降低使用门槛。
基于码本查找 Transformer 的盲人脸修复工具,可增强模糊、低质量或受损的人脸图像与视频。