YOLOX: Anchor-free object detection model
PyTorch anchor-free object detector with scalable model variants.
Learn more about YOLOX
YOLOX is an object detection model that removes the anchor-based detection mechanism used in earlier YOLO versions, replacing it with an anchor-free approach. The architecture is implemented in PyTorch with a companion MegEngine implementation available separately. The model family includes variants ranging from YOLOX-Nano with 0.91M parameters to YOLOX-X with 99.1M parameters, enabling deployment across different computational constraints. Common applications include real-time object detection in computer vision pipelines, with support for edge deployment through quantization and conversion to optimized inference formats.
Anchor-Free Detection
Eliminates anchor boxes from the detection pipeline, removing hyperparameter tuning for anchor dimensions and aspect ratios. Simplifies training workflow while maintaining accuracy comparable to anchor-based YOLO versions.
Multi-Framework Deployment
Export trained models to ONNX, TensorRT, ncnn, OpenVINO, and MegEngine with built-in conversion scripts. Enables deployment across cloud GPUs, edge devices, and mobile hardware without manual optimization.
Scalable Model Variants
Provides five model sizes from Nano (0.91M parameters) to X (99.1M parameters) with consistent architecture. Allows selection based on target hardware constraints, from resource-limited embedded devices to high-performance server inference.
import torch
from yolox.exp import get_exp
exp = get_exp(None, "yolox-s")
model = exp.get_model()
ckpt = torch.load("yolox_s.pth", map_location="cpu")
model.load_state_dict(ckpt["model"])
model.eval()
img = torch.randn(1, 3, 640, 640)
outputs = model(img)YOLOX 0.3.0
- –support loading YOLOX model through `torch.hub` #1189
- –support just-in-time compile op #1241
- –support wandb logger #1144
- –support `freeze` function for torch module #1156
- –support showing YOLOX demo in a live window #1138
```pip install yolox``` could help you to install yolox now.
- –Saved 30% memory useage in COCO training. #1066
- –Log per class AP & AP during evaluation. #1026 #1052
- –Users could install yolox from pip now! Supports on more platform is coming. #1020 #1079
- –Optimize dynamic matching in label assignment. #861
0.1.1 pre release
- –Support image caching for faster training, which requires large system RAM.
- –Remove the dependence of apex and support torch amp training.
- –Optimize the preprocessing for faster training
- –Replace the older distort augmentation with new HSV aug for faster training and better performance.
- –d: number of gpu devices
See how people are using YOLOX
Top in AI & ML
Related Repositories
Discover similar tools and frameworks used by developers
docling
Fast document parser for RAG and AI workflows.
unsloth
Memory-efficient Python library for accelerated LLM training.
pytorch
Python framework for differentiable tensor computation and deep learning.
nanoGPT
Minimal PyTorch implementation for training GPT models.
presentation-ai
AI-powered slide generator with multi-model integration and themes.