def track(frame, avg_frame):
gray = cv2.cvtColor(frame, cv2.COLOR_RGB2GRAY)
gray_copy = gray.copy()
img_threshold, x, y = ta.diffaccWeight(gray_copy, gray, avg_frame)
return img_threshold, x, y
got_frame0, frame0 = myCamera0.getFrame() image0 = cv2.resize(frame0,None,fx=0.25, fy=0.25, interpolation = cv2.INTER_LINEAR) t0 = cv2.cvtColor(image0, cv2.COLOR_RGB2GRAY) got_frame1, frame1 = myCamera1.getFrame() image1 = cv2.resize(frame1,None,fx=0.25, fy=0.25, interpolation = cv2.INTER_LINEAR) t1 = cv2.cvtColor(image1, cv2.COLOR_RGB2GRAY) # t0 = cv2.cvtColor(frame0, cv2.COLOR_RGB2GRAY) # t1 = cv2.cvtColor(frame1, cv2.COLOR_RGB2GRAY) f0 = t0.copy() f1 = t1.copy() masked_img0, x0, y0 = ta.diffaccWeight(f0,t0, avg_daw0) masked_img1, x1, y1 = ta.diffaccWeight(f1,t1, avg_daw1) if((x0 != -1) | (y0 != -1)): measured0 = (x0,y0) prev_x0 = x0 prev_y0 = y0 else: measured0 = (prev_x0,prev_y0) if((x1 != -1) | (y1 != -1)): measured1 = (x1,y1) prev_x1 = x1 prev_y1 = y1 else:
myCamera0 = Camera.Camera(0, "cam0.avi")
got_frame, frame = myCamera0.getFrame()
t0 = cv2.cvtColor(frame, cv2.COLOR_RGB2GRAY)
avg_daw0 = np.float32(t0)
i = 0
start_time = time.time()
while True:
got_frame, frame = myCamera0.getFrame()
t0 = cv2.cvtColor(frame, cv2.COLOR_RGB2GRAY)
f0 = t0.copy()
masked_img0, x0, y0 = ta.diffaccWeight(f0,t0, avg_daw0)
#if((x0 != -1) | (y0 != -1)):
drawCross(masked_img0, (x0, y0), 0, 0, 255)
cv2.imwrite('frame' + str(i) + ".jpg", masked_img0)
cv2.imshow( "frame", masked_img0 )
i+=1
if (i == 30):
#finish_time = time.time()
break
key = cv2.waitKey(1)
if key == 27:
cv2.destroyWindow("0")
myCamera0.off()
break
def track(frame, avg_frame): gray = cv2.cvtColor(frame, cv2.COLOR_RGB2GRAY) gray_copy = gray.copy() img_threshold, x, y = ta.diffaccWeight(gray_copy,gray, avg_frame) return img_threshold, x, y
