Exemplo n.º 1
0
# 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: