Awesome
YOLO for Intel/Movidius Neural Compute Stick (NCS)
News
- Camera App is working.
- YOLOv1 Tiny is working.
Protobuf Model files
./prototxt/
Download Pretrained Caffe Models to ./weights/
Compilation
- Compile .prototxt and corresponding .caffemodel (with the same name) to get NCS graph file. For example: "mvNCCompile prototxt/yolo_tiny_deploy.prototxt -w weights/yolo_tiny_deploy.caffemodel -s 12"
- The compiled binary file "graph" has to be in main folder after this step.
Single Image Script
- Run "yolo_example.py" to process a single image. For example: "python3 py_examples/yolo_example.py images/dog.jpg" to get detections as below.
Camera Input Script
- Run "object_detection_app.py" to process a videos from your camera. For example: "python3 py_examples/object_detection_app.py" to get camera detections as below.
- Modify script arguments if needed.
- Press "q" to exit app.