- OpenCV 3.4
- imutils
Download weights here and place them in model_data/
$python3 src/main.py -h
usage: main.py [-h] [--input INPUT] [--output OUTPUT] --model MODEL
[--config CONFIG] [--classes CLASSES] [--thr THR]
Object Detection and Tracking on Video Streams
optional arguments:
-h, --help show this help message and exit
--input INPUT Path to input image or video file. Skip this argument to
capture frames from a camera.
--output OUTPUT Path to save output as video file. Skip this argument if
you don't want the output to be saved.
--model MODEL Path to a binary file of model that contains trained weights.
It could be a file with extensions .caffemodel (Caffe) or
.weights (Darknet)
--config CONFIG Path to a text file of model that contains network
configuration. It could be a file with extensions
.prototxt (Caffe) or .cfg (Darknet)
--classes CLASSES Optional path to a text file with names of classes to
label detected objects.
--thr THR Confidence threshold for detection. Default: 0.35
Execute code from root directory. Example:
python3 src/main.py --model model_data/yolov2.weights --config model_data/yolov2.cfg --classes model_data/coco_classes.txt --input media/sample_video.mp4 --output out/sample_output.avi
or
python3 src/main.py --model model_data/MobileNetSSD_deploy.caffemodel --config model_data/MobileNetSSD_deploy.prototxt --classes model_data/MobileNet_classes.txt --input media/sample_video.mp4 --output out/sample_output.avi
Note: --input can be ommitted, which will activate stream from webcam. New objects are detected when all current objects being tracked are lost, or when 'q' is pressed
