← all repositories

Tianxiaomo/pytorch-YOLOv4

A minimal PyTorch implementation of YOLOv4 object detection with support for training, ONNX export, and TensorRT inference.

4.5k stars Python Computer VisionML Frameworks
pytorch-YOLOv4
Velocity · 7d
+2.0
★ / day
Trend
steady
star history

This repository provides a PyTorch implementation of the YOLOv4 object detection model originally from the Darknet framework. It includes utilities for training the model on custom datasets, converting Darknet weights to PyTorch format, exporting models to ONNX for cross-platform deployment, and optimizing inference with TensorRT for GPU acceleration. The implementation supports standard YOLOv4 and YOLOv4-tiny variants.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.