bit_frame=np.zeros((old_frame.shape[0],old_frame.shape[1]),dtype=np.bool) bitmask=np.zeros((old_frame.shape[0],old_frame.shape[1]),dtype=np.uint8) out_frame=np.zeros((frame_height,frame_width,num_channels),dtype=np.uint8) t=0 def get_new_frame(frame,nlevs): band=frame for lev in range(nlevs): band=cv2.pyrDown(band) return band #first frame cap.set(cv2.CAP_PROP_POS_FRAMES,t); ret,frame=cap.read() pyr_blender=pyrMask.pyrMask(frame,True,nlevs,inference_lev) old_frame[:,:,:]=get_new_frame(frame,inference_lev) started=False tle=0 print tot_time t = t+1 while(cap.isOpened() and ret): print t cap.set(cv2.CAP_PROP_POS_FRAMES,skip_frames*t); ret, frame = cap.read() new_frame[:,:,:]=get_new_frame(frame,inference_lev) tic=time.time() corr_frame[:,:]=corrConv.fastCorr(old_frame,new_frame,corr_support)
new_frame=np.zeros((frame_height,frame_width,num_channels),dtype=np.int16) mask_frame=np.zeros((frame_height,frame_width,num_channels),dtype=np.uint8) last_frame=np.zeros((frame_height,frame_width,num_channels),dtype=np.uint8) diff_frame=np.zeros((frame_height,frame_width,num_channels),dtype=np.int16) vote_frame=np.zeros((frame_height,frame_width),dtype=np.float32) bool_frame=np.zeros((frame_height,frame_width),dtype=np.bool) bit_frame=np.zeros((frame_height,frame_width),dtype=np.bool) erode_frame=np.zeros((frame_height,frame_width),dtype=np.float32) dilate_frame=np.zeros((frame_height,frame_width),dtype=np.float32) out_frame=np.zeros((frame_height,frame_width,num_channels),dtype=np.uint8) time=skip_ahead #first frame cap.set(cv2.CAP_PROP_POS_FRAMES,skip_frames*time); ret,frame=cap.read() pyr_blender=pyrMask.pyrMask(frame,True) old_frame[:,:,:]=frame started=False up=True old_tle=pyr_blender.get_top_level_energy() new_tle=0 print old_tle time = time+1 best_score=255*num_channels while(cap.isOpened() and ret): print time cap.set(cv2.CAP_PROP_POS_FRAMES,skip_frames*time); ret, frame = cap.read() new_frame[:,:,:]=frame diff_frame[:,:,:]=np.abs(old_frame-new_frame) med=np.median(diff_frame)
old_frame = np.zeros((frame_height, frame_width, num_channels), dtype=np.uint8) new_frame = np.zeros((frame_height, frame_width, num_channels), dtype=np.uint8) mask_frame = np.zeros((frame_height, frame_width, num_channels), dtype=np.uint8) corr_frame = np.zeros((frame_height, frame_width), dtype=np.float) bit_frame = np.zeros((frame_height, frame_width), dtype=np.bool) bool_frame = np.zeros((frame_height, frame_width), dtype=np.bool) bitmask = np.zeros((frame_height, frame_width), dtype=np.uint8) out_frame = np.zeros((frame_height, frame_width, num_channels), dtype=np.uint8) t = 0 # first frame cap.set(cv2.CAP_PROP_POS_FRAMES, t) ret, frame = cap.read() pyr_blender = pyrMask.pyrMask(frame, True, nlevs, 0) old_frame[:, :, :] = frame started = False tle = 0 print tot_time t = t + 1 while cap.isOpened() and ret: print t cap.set(cv2.CAP_PROP_POS_FRAMES, skip_frames * t) ret, frame = cap.read() new_frame[:, :, :] = frame tic = time.time() corr_frame[:, :] = corrConv.fastCorr(old_frame, new_frame, corr_support) print time.time() - tic bitmask[:, :] = np.uint8(corr_frame > confidence) * 255