def write_video(frames, out_vid, background_vid, start_frame, end_frame): fps = 7 capSize = (1920, 1080) # this is the size of my source video codec_x264 = cv2.VideoWriter_fourcc('X', '2', '6', '4') out = cv2.VideoWriter(out_vid, codec_x264, fps, capSize, True) if background_vid is not None: cap = cv2.VideoCapture(background_vid) counter = 1 while (cap.isOpened()): ret, color_frame = cap.read() if not ret: print('Cannot open video file!') break else: #print(counter) if (counter - start_frame == len(frames)): break if counter < start_frame: counter += 1 continue frame = frames[counter - start_frame] resized_frame = cv2.resize(color_frame, (1920, 1080)) out.write(cv2.add(drawing.drawframe(frame), resized_frame)) counter += 1 cap.release() else: for frame in frames: out.write(drawing.drawframe(frame)) out.release()
def write(frames, path, name='/test.avi', overColor=True): global FPS capSize = (1920, 1080) # this is the size of my source video codec_x264 = cv2.VideoWriter_fourcc('X', '2', '6', '4') out = cv2.VideoWriter(path + name, codec_x264, FPS, capSize, True) if (overColor): cap = cv2.VideoCapture(path + '/color.avi') counter = 0 while (cap.isOpened()): ret, color_frame = cap.read() if not ret: print('Cannot open video file!') break else: #print(counter) if (counter == len(frames)): break frame = frames[counter] resized_frame = cv2.resize(color_frame, (1920, 1080)) out.write(cv2.add(drawing.drawframe(frame), resized_frame)) counter += 1 cap.release() else: for frame in frames: out.write(drawing.drawframe(frame)) out.release()
def write_video(frames, out_vid, background_vid, start_frame, end_frame, fps, color_inferred, color_tracked): capSize = (1920, 1080) # this is the size of my source video #codec_x264 = cv2.VideoWriter_fourcc('X', '2', '6', '4') codec_x264 = cv2.VideoWriter_fourcc(*'MJPG') out = cv2.VideoWriter(out_vid, codec_x264, fps, capSize, True) if background_vid is not None: cap = cv2.VideoCapture(background_vid) counter = 1 while (cap.isOpened()): ret, color_frame = cap.read() if not ret: if counter == 1: print('Cannot open video file!') break else: if (counter - start_frame == len(frames)): break if counter < start_frame: counter += 1 continue frame = frames[counter - start_frame] resized_frame = cv2.resize(color_frame, (1920, 1080)) #print("frame") #print(len(frame)) im = drawing.drawframe(frame, color_inferred, color_tracked) #print(im.shape) #print(resized_frame.shape) #out.write(cv2.add(drawing.drawframe(frame, color_inferred, color_tracked),resized_frame)) out.write(cv2.add(im, resized_frame)) counter += 1 #print(counter) cap.release() else: for frame in frames: out.write(drawing.drawframe(frame, color_inferred, color_tracked)) out.release()
def draw_frames(frames): cv2.namedWindow('skeleton') for i, frame in enumerate(frames): img = drawing.drawframe(frame) cv2.imshow('skeleton', img) cv2.waitKey(100)