erentknn/yolov4-object-detection

Simple inference script for YOLOv4 using opencv.

Python

Simple Inference Scripts for YOLO with OpenCV

I made a YOLOv4 inference script for both video and image files, easy to use and not complicated.

Example Usage

python yolo_image.py -i street.jpg -o output.jpg

python yolo_video.py -i video.mp4 -o video_out.avi

TODO

Add custom training script

References

https://github.com/opencv/opencv/blob/master/samples/dnn/object_detection.py https://www.pyimagesearch.com/2020/06/01/opencv-social-distancing-detector/