Skip to content

miladlink/TinyYoloV2

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Pytorch Tiny YoloV2 implementation from scratch

Description

  • YOLO or You Only Look Once, is a popular real-time object detection algorithm.

  • YOLO combines what was once a multi-step process, using a single neural network to perform both classification and prediction of bounding boxes for detected objects.

  • network divides the image into regions and predicts bounding boxes and probabilities for each region.

image

  • In this repo implemented one of the simple and fast version of YOLO from scratch.

Dataset

  • Pascal-VOC dataset is the one of the collections for object detection

  • Pretrained weights in this implemetation are based on training yolo team on VOC dataset

Pretrained Weights

  • You can check the yolo website for defferent variation pretrained weights for different sizes or datasets.

  • Download used weights in this project from here

  • Loading this weights is not like conventional methods (like loading .pth, .pt, ... formats) so they should put on the model's body

Result images

  • Predictions and bounding boxes not so much accurate but it's fast

person

dog

horses

Note: predicted objects is around defined class names so it can not predict out of this such as below image

giraffe