time=time+1 if time*skip_frames >= tot_time: break time=time-nD-2 final_label=np.array(np.argmin(old_energy,1),dtype=np.uint8) prev_labels=np.zeros(num_pixels,dtype=np.uint8) prev_labels[:]=labels[range(num_pixels),final_label] #outframe=cv2.cvtColor(curr_data.get_output(final_label),3) outframe=np.zeros(frame.shape,dtype=np.uint8) outframe[:,:,:]=curr_data.get_output(final_label) inframe=np.zeros(frame.shape,dtype=np.uint8) print curr_data.now.shape inframe[:,:,:]=np.reshape(curr_data.now,frame.shape) vid_cache=pyrStripe.pyrStripe(outframe) writeframe=np.zeros(frame.shape,dtype=np.uint8) writeframe[:,:,:]=outframe out.write(outframe) for t in range(time,-1,-1): labels[:,:]=bp[:,:,t+nD] curr_data.add_frame_left() outframe[:,:,:]=curr_data.get_output(prev_labels) inframe[:,:,:]=np.reshape(curr_data.now,frame.shape) (a,b)=get_dispute(inframe,outframe) writeframe[:,:,:]=vid_cache.stripeOut(inframe,a,b) out.write(writeframe) prev_labels[:]=labels[range(num_pixels),prev_labels] print t
import cv2 import numpy as np import pyrStripe import lPyr import signal in_name='../subsamp-change-threshold-0.06-MH12non-euclid.avi' cap = cv2.VideoCapture(in_name) cap.set(cv2.CAP_PROP_POS_FRAMES,4) ret, frame4 = cap.read() cap.set(cv2.CAP_PROP_POS_FRAMES,100) ret, frame100 = cap.read() n=pyrStripe.pyrStripe(frame4) p=n.stripeOut(frame100,200,800) #cv2.imshow('p',p) #print 't' def preent(): print 't' def iirr(signal,r): y=np.zeros(signal.shape[0],dtype=np.int64) z=signal[0] y[0]=z for n in range(1,signal.shape[0]): z=z*r+signal[n]*(1-r) y[n]=z return y def rms(signal,support): return signal def battle(signal,start):