yolo

yolo.py

@@ -6,6 +6,7 @@ import time
 parser = argparse.ArgumentParser()
 parser.add_argument('--webcam', help="True/False", default=False)
 parser.add_argument('--play_video', help="Tue/False", default=False)
+parser.add_argument('--play_flask', help="Tue/False", default=False)
 parser.add_argument('--image', help="Tue/False", default=False)
 parser.add_argument('--video_path', help="Path of video file", default="videos/car_on_road.mp4")
 parser.add_argument('--image_path', help="Path of image to detect objects", default="Images/bicycle.jpg")
@@ -82,7 +83,20 @@ def draw_labels(boxes, confs, colors, class_ids, classes, img):
             color = colors[i]
             cv2.rectangle(img, (x,y), (x+w, y+h), color, 2)
             cv2.putText(img, label, (x, y - 5), font, 3, color, 3)
-    cv2.imshow("Image", img)
+    imS = cv2.resize(img, (960, 540))
+    cv2.imshow("Image", imS)
+
+def draw_labels2(boxes, confs, colors, class_ids, classes, img):
+    indexes = cv2.dnn.NMSBoxes(boxes, confs, 0.4, 0.4)
+    font = cv2.FONT_HERSHEY_PLAIN
+    for i in range(len(boxes)):
+        if i in indexes:
+            x, y, w, h = boxes[i]
+            label = str(classes[class_ids[i]])
+            color = colors[i]
+            cv2.rectangle(img, (x,y), (x+w, y+h), color, 2)
+            cv2.putText(img, label, (x, y - 5), font, 3, color, 3)
+    return img
 
 def image_detect(img_path):
     model, classes, colors, output_layers = load_yolo()
@@ -119,17 +133,57 @@ def start_video(video_path):
         blob, outputs = detect_objects(frame, model, output_layers)
         boxes, confs, class_ids = get_box_dimensions(outputs, height, width)
         draw_labels(boxes, confs, colors, class_ids, classes, frame)
-        k = cv2.waitKey(1) & 0xFF
+        k = cv2.waitKey(50) & 0xFF
         if k == 27:
             cv2.destroyAllWindows()
             break
     cap.release()
 
 
+from flask import Flask, render_template, Response
+
+app = Flask(__name__)
+def find_camera(id):
+    '''
+    cameras = ['rtsp://username:password@ip_address:554/user=username_password='password'_channel=channel_number_stream=0.sdp',
+    'rtsp://username:password@ip_address:554/user=username_password='password'_channel=channel_number_stream=0.sdp']
+    '''
+    cameras = ['rtsp://admin:@Unv123456@192.168.10.252:554/unicast/c1/s1/live']
+    return cameras[int(id)]
+#  for cctv camera use rtsp://username:password@ip_address:554/user=username_password='password'_channel=channel_number_stream=0.sdp' instead of camera
+#  for webcam use zero(0)
+def gen_frames(camera_id):
+
+    cam = find_camera(camera_id)
+    video_path = cam
+    model, classes, colors, output_layers = load_yolo()
+    cap = cv2.VideoCapture(video_path)
+    while True:
+        _, frame = cap.read()
+        if frame is not None:
+            height, width, channels = frame.shape
+            blob, outputs = detect_objects(frame, model, output_layers)
+            boxes, confs, class_ids = get_box_dimensions(outputs, height, width)
+            frame = draw_labels2(boxes, confs, colors, class_ids, classes, frame)
+            ret, buffer = cv2.imencode('.jpg', frame)
+            frame = buffer.tobytes()
+            yield (b'--frame\r\n'
+                b'Content-Type: image/jpeg\r\n\r\n' + frame + b'\r\n')  # concat frame one by one and show result
+    cap.release()
+
+
+@app.route('/video_feed/<string:id>/', methods=["GET"])
+def video_feed(id):
+
+    """Video streaming route. Put this in the src attribute of an img tag."""
+    return Response(gen_frames(id),
+                    mimetype='multipart/x-mixed-replace; boundary=frame')
+
 
 if __name__ == '__main__':
     webcam = args.webcam
     video_play = args.play_video
+    flask_play = args.play_flask
     image = args.image
     if webcam:
         if args.verbose:
@@ -146,5 +200,6 @@ if __name__ == '__main__':
             print("Opening "+image_path+" .... ")
         image_detect(image_path)
 
-
+    if flask_play:
+        app.run(debug=True, port=9024)
     cv2.destroyAllWindows()