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
