From 3a0aad1549a70009311d46a0802998e07d0735e1 Mon Sep 17 00:00:00 2001 From: au Date: Sat, 2 Dec 2017 09:21:52 +0200 Subject: [PATCH] added ability to webcam to predict from --- predict.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/predict.py b/predict.py index 4e6727d75..8f5b03290 100644 --- a/predict.py +++ b/predict.py @@ -84,7 +84,17 @@ def _main_(args): video_writer.write(np.uint8(image)) video_reader.release() - video_writer.release() + video_writer.release() + elif image_path == 'cam.0': + cam = cv2.VideoCapture(0) + while True: + ret_val, image = cam.read() + boxes = yolo.predict(image) + image = draw_boxes(image, boxes, config['model']['labels']) + cv2.imshow('my webcam', image) + if cv2.waitKey(1) == 27: + break + cv2.destroyAllWindows() else: image = cv2.imread(image_path) boxes = yolo.predict(image)