Tianxiaomo/pytorch-YOLOv4

Health v2 · maintenance only

Flags: no_releases

Adoption not part of the score

4521 stars · 1481 forks observed · 2026-08-28

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

A minimal PyTorch implementation of YOLOv4 (and YOLOv4-tiny) supporting inference and training, with tools to convert Darknet weights to PyTorch and ONNX, and ONNX to TensorRT. It includes pretrained weights, training scripts with Mosaic augmentation, and demo utilities.

Use cases

• run YOLOv4 object detection inference in PyTorch
• train YOLOv4 on my own custom dataset
• convert darknet weights to pytorch
• convert YOLOv4 model to ONNX for deployment
• deploy YOLOv4 with TensorRT for faster inference
• use yolov4-tiny for lightweight detection
• detect objects in images or camera feed

When to choose

• you need YOLOv4 in PyTorch rather than Darknet
• you want to fine-tune YOLOv4 on custom data with Mosaic augmentation
• you need a conversion path from Darknet weights to ONNX/TensorRT

When to avoid

• you need the latest YOLO architectures like YOLOv7/v8 or YOLO11
• you want a maintained production pipeline with active development
• you need non-PyTorch frameworks like TensorFlow

Facets

library · maturity maintenance

machine-learning computer-vision image-processing deep-learning computer-vision machine-learning deep-learning image-processing python cross-platform yolov4 object-detection pytorch onnx tensorrt darknet yolov4-tiny model-conversion gpu

Member repositories

For agents

markdown · JSON · MCP: product_card(name="Tianxiaomo/pytorch-YOLOv4")