bubbliiiing/yolov5-pytorch

Health v2 · maintenance only

Adoption not part of the score

1149 stars · 186 forks observed · 2026-08-28

What it is AI-extracted, prompt v1, taxonomy v1, 2026-08-30, confidence not recorded

A PyTorch implementation of the YOLOv5 (v5.0) object detection model with full training, prediction, and evaluation pipelines. It is designed for training custom object detection models on user-provided datasets such as VOC-format annotations.

Use cases

• train a custom object detection model on my own dataset
• run YOLOv5 object detection on images in PyTorch
• convert VOC annotations and train YOLOv5 s/m/l/x variants
• evaluate object detection mAP on COCO or VOC datasets
• learn how YOLOv5 works internally with readable source code

When to choose

• you want a clean, educational PyTorch YOLOv5 codebase to train custom models
• you need full-precision weights and step-by-step training scripts in Chinese and English
• you want multi-GPU training support with configurable optimizers and learning rate schedules

When to avoid

• you need the latest Ultralytics YOLOv5/v6.1 features or production-grade tooling
• you require commercial use under a permissive license (it is GPL-3.0)
• you need active ongoing development or frequent updates

Facets

library · maturity maintenance

machine-learning deep-learning computer-vision image-processing computer-vision deep-learning machine-learning image-processing python cross-platform yolov5 object-detection pytorch model-training custom-dataset gpu

Member repositories

For agents

markdown · JSON · MCP: product_card(name="bubbliiiing/yolov5-pytorch")