# find the top velocity events maxgap=15 groups = cluster(frameset,maxgap) framegroups = [] ############################################ # test section to write out frames as videos # extract the frames to be operated on framegroupstest = [] videouttest = [] for set in groups: if len(set) >= setnumber: # process the video again extracting the relevant segments # input arguments are the frame numbers and location of the object in # the frame. segment = frameExtractor.extractVideoSegment(args["video"], set, framesetdata, framespre, framespost, xsize, ysize) framegroupstest.append(segment) counter = 0 for frameg in framegroupstest: # write to a file counter = counter + 1 voname= "tracktest" + str(counter) + ".avi" fourcc_pre = cv2.VideoWriter_fourcc(*'XVID') outvideo = cv2.VideoWriter(voname, fourcc_pre, fps, (xproc,yproc)) # open output video to record for frame in frameg: outvideo.write(frame) outvideo.release() # new write just the segment for this set framegroupstest = [] videouttest = [] for set in groups: