/YOLOv1-from-scratch

YOLOv1-from-scratch

Primary LanguagePython

README

implementation of YOLOv1

from BLOG

Required:

  • pytorch 1.1.0
  • torchvision
  • numpy > 1.16.2
  • opencv 3.4.1
  • VOC2012 Dataset

What can this repo do now

  • train with VOC2012 Dataset
  • inference(test images)

Result

image-20210327092525477

2007_000032

2007_000033

TODO:

  • add mAP metric