Project Title

Object Detection with YOLOv1

Description

Recreated the YOLOv1 algorithm for object detection, trained on the MS COCO Dataset.
Guided by Aladdin Persson's "YOLOv1 from Scratch" video (https://youtu.be/n9_XyCGr-MI?si=NUCf7BkqcFDk022D)

Install Pytorch (https://pytorch.org/get-started/locally/)
Download COCO Dataset 2017 Train Images, 2017 Val Images, and 2017 Train/Val Annotations (https://cocodataset.org/#download)

Change file directory to COCO Dataset images and annotations in train.py
Run train.py

Name		Name	Last commit message	Last commit date
Latest commit History 7 Commits
README.md		README.md
dataset.py		dataset.py
loss.py		loss.py
model.py		model.py
train.py		train.py
utils.py		utils.py