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

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.