This repo contains my implementations of popular object detection algorithms in Pytorch (more specifically using Pytorch-Lighning). I try to implement them as I learn them, feel free create an issue if you find any bug in my code.
Implemented Algorithms so far:
- Yolo V1 (https://arxiv.org/abs/1506.02640)
- Yolo V2 (https://arxiv.org/abs/1612.08242)
- Yolo V3 (https://arxiv.org/abs/1804.02767)
- Yolo V4
- Yolo V5
- SSD
- RetinaNet
- EfficientDet