play.py

import cv2
from cv import predictor
from motion_detection import motion
import numpy as np
#laptop
# cl = cv2.VideoCapture(0)
#leon
#c = cv2.VideoCapture('http://100.65.194.194:4747/mjpegfeed')
#ad
# c = cv2.VideoCapture('http://100.64.228.178:4747/mjpegfeed')
cameras = [cv2.VideoCapture(0), cv2.VideoCapture('http://100.64.228.178:4747/mjpegfeed'), cv2.VideoCapture('http://100.65.194.194:4747/mjpegfeed')]
# while(1):
#     _,f = c.read()
#     cv2.imshow('e2',f)
#     _, fl = cl.read()
#     cv2.imshow('e3', fl)
#     if cv2.waitKey(5)==27:
#         break
# cv2.destroyAllWindows()
p1 = predictor.Predictor()
p2 = motion.Predictor(len(cameras))
# predictor.Algorithms.init()
# frame_interval = predictor.runDiagnostic()
counter = 0
lastf = 0
while (1):
    # ret, frame = cap.read()
    frames = [0]*len(cameras)
    for i in range(0, len(cameras)):
        print(i)
        _, frames[i] = cameras[i].read()
    
    #_, f = .read()
    #cv2.imshow('e2', f)
    #_, fl = cl.read()
    #cv2.imshow('e3', fl)
    # gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
    # resized_gray = imutils.resize(gray, width=500)
    # cv2.imshow('frame', resized_gray)
    counter += 1
    #print("here")
    #fs = [fl, f]
    #cv2.imshow("stream", lastf)
    if counter % 3 == 0:
        ret1 = p1.next(frames)
        ret2 = p2.next(frames)
        if ret2 == []:
            ret2 = [0] * len(cameras)
        ret = [0] * len(cameras)
        for i in range(0, len(cameras)):
            ret[i] = 0.7 * ret1[i] + 0.3 * ret2[i]
        # ret = ret2
        if max(ret) == 0:
            print("ZERO")
            cv2.imshow("stream", frames[lastf])
        else:
            best = frames[ret.index(max(ret))]
            cv2.imshow("stream", best)
            lastf = ret.index(max(ret))
        print(ret)
        print(ret.index(max(ret)))
        #mouths = predictor.getTestingImage(f)
        #mouths2 = predictor.getTestingImage(fl)
        counter = 0
    else:
        cv2.imshow("stream", frames[lastf])
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break
    #cv2.imshow('color', mouths)
    #cv2.imshow('color2', mouths2)



# cap.release()