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)