Skip to content

kalyco/yolo_detector

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

14 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Yolo Detector Class

Image and Video detector class functions based off pyimagesearch and yolo

Requirements

  • Python 3.5
  • OpenCV
  • A Yolo weight file (too big for github)

Running Detectors

Image Detector

ex: python yolo_image.py --image images/baggage_claim.jpg --yolo yolo-coco

Command Shortcut Description Required Default
--image -image path to image True
--yolo -y base path to YOLO directory True
--confidence -c minimum probability to filter weak detections 0.5
--threshold -t threshold when applying non-maxima suppression 0.3

Video Detector

Command Shortcut Description Required Default
--input -i path to input video" True
--output -o path to output video" True
--yolo -y base path to YOLO directory True
--confidence -c minimum probability to filter weak detections 0.5
--threshold -t threshold when applying non-maxima suppression 0.3

ex: python yolo_video.py --input videos/airport.mp4 --output output/airport_output.avi --yolo yolo-coco

alt text

About

Yolo object detector as a class

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages