← all repositories

ruiminshen/yolo2-pytorch

PyTorch implementation of the YOLOv2 deep learning object detection model.

442 stars Python Computer VisionML Frameworks
yolo2-pytorch
Velocity · 7d
+0.1
★ / day
Trend
steady
star history

This repository provides a PyTorch implementation of the YOLO v2 object detection algorithm, a popular one-stage detector. It includes training pipelines with TensorBoard monitoring, support for ONNX model export for Caffe2 deployment, configurable settings via command line, and checkpoint management with NaN debugging capabilities.

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