Skip to content

Latest commit

 

History

History
13 lines (9 loc) · 580 Bytes

README.md

File metadata and controls

13 lines (9 loc) · 580 Bytes
Raw

YOLO

My PyTorch implementation of the YOLO family of algorithms following the publications of Joseph Redmon and others. Currently implemented are:

  • YOLO_V1 [Redmon et.al, 2015]
  • YOLO_V2/3 ... work in progress

My motivation behind this project was to get a better understanding of how the loss function for the YOLO family of algorithm is designed and how they can be implemented in PyTorch efficiently.

Datasets

For training and evaluation I use the 8- and 100-example versions of the COCO dataset, which are also used in YOLO_V1.