← all repositories

abeardear/pytorch-YOLO-v1

A PyTorch implementation of YOLO-v1 object detection using ResNet50 backbone.

673 stars Python Computer Vision
pytorch-YOLO-v1
Velocity · 7d
+0.2
★ / day
Trend
steady
star history

This repository contains an experimental implementation of the YOLO-v1 object detection algorithm in PyTorch. The author replaced the original darknet backbone with ResNet50 to leverage pretrained weights. The code supports training on VOC2007+2012 datasets and evaluation, achieving 66.5% mAP at 57 FPS on a GTX 1080 Ti. It includes loss computation using autograd, data preparation scripts, and inference visualization.

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