vhad/body-pose-animation
1
0
Fork 0
mirror of https://github.com/gosticks/body-pose-animation.git synced 2025-10-16 11:45:42 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

add mp4 format

Browse Source
This commit is contained in:
Wlad 2021-02-09 10:47:20 +01:00
parent 93d6822b6a
commit 6ee7a30855
1 changed files with 12 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
utils/video.py
View File

@ -6,12 +6,19 @@ from tqdm import tqdm
import numpy as np import numpy as np
def make_video(images, video_name: str, fps=5): def make_video(images, video_name: str, fps=5, ext: str = "mp4"):
images = np.array(images) images = np.array(images)
width = images.shape[2] width = images.shape[2]
height = images.shape[1] height = images.shape[1]
fourcc = 0
if ext == "mp4":
fourcc = cv2.VideoWriter_fourcc(*'MP4V')
video_name = video_name + "." + ext
video = cv2.VideoWriter( video = cv2.VideoWriter(
video_name, 0, fps, (width, height), True) video_name, fourcc, fps, (width, height), True)
print("creating video with size", width, height) print("creating video with size", width, height)
@ -23,11 +30,10 @@ def make_video(images, video_name: str, fps=5):
video.release() video.release()
def video_from_pkl(filename, video_name, config): def video_from_pkl(filename, video_name, config, ext: str = "mp4"):
with open(filename, "rb") as fp: with open(filename, "rb") as fp:
final_poses = pickle.load(fp) model_outs = pickle.load(fp)
save_to_video(model_outs, video_name, config)
save_to_video(final_poses, video_name, config)
def save_to_video(poses, video_name, config, fps=30): def save_to_video(poses, video_name, config, fps=30):