Skip to content

Instantly share code, notes, and snippets.

@vinooniv
vinooniv / opencv_yolo_detector.py
Created November 15, 2019 15:49
YOLO with OpenCV DNN module
import argparse
import cv2
import numpy as np
parser = argparse.ArgumentParser(add_help=False)
parser.add_argument("--image", default='samples/image.jpg', help="image for prediction")
parser.add_argument("--config", default='cfg/yolov3.cfg', help="YOLO config path")
parser.add_argument("--weights", default='yolov3.weights', help="YOLO weights path")
parser.add_argument("--names", default='data/coco.names', help="class names path")