finds.dev← search

// the find

bubbliiiing/yolov5-pytorch

★ 1,149 · Python · GPL-3.0 · updated Aug 2023

这是一个YoloV5-pytorch的源码,可以用于训练自己的模型。

A clean re-implementation of YOLOv5 v5.0 in PyTorch, written by a Chinese developer who has done similar repos for v3, v4, v7, and YoloX. Primarily a learning and fine-tuning resource — if you want to understand the architecture or train a custom detector without fighting Ultralytics' opinionated CLI, this is readable code. Documentation is Chinese-only.

The codebase is genuinely readable — model components in nets/ are split logically (CSPDarknet, head, training loss), not buried in one monolithic file. It includes ConvNext and Swin Transformer backbone variants, which the stock Ultralytics v5 didn't ship with. Multi-GPU training, EMA, and cosine LR decay are all present and functional. Performance numbers match the official release closely (mAP 0.5 of 53.9 for YOLOv5-s vs Ultralytics' 56.8 — difference is full vs half precision weights).

Frozen at v5.0 from 2022; last commit August 2023, so you're three major Ultralytics releases behind and won't get any of the v6/v7 head improvements. Requires torch==1.2.0 in the README which is almost certainly wrong (that's a 2019 build) — the actual requirements.txt probably says something saner but it's a red flag for documentation accuracy. All issues, comments, and the companion BiliBili tutorial are in Chinese, so debugging community support is limited if you don't read Mandarin. No ONNX or TensorRT export path documented, which is where most production deployments actually go.

View on GitHub →

// want more like this?

We dig through GitHub every week and send a few repos picked for what you actually care about — each with an honest take like this one.

Get finds in your inbox → Search again →