|
import os |
|
import pickle |
|
import random |
|
import sys |
|
import natsort |
|
from argparse import ArgumentParser |
|
|
|
try: |
|
import bpy |
|
sys.path.append(os.path.dirname(bpy.data.filepath)) |
|
except ImportError: |
|
raise ImportError( |
|
"Blender is not properly installed or not launch properly. See README.md to have instruction on how to install and use blender.") |
|
|
|
import mld.launch.blender |
|
from mld.render.blender import render |
|
|
|
|
|
def parse_args(): |
|
parser = ArgumentParser() |
|
parser.add_argument("--pkl", type=str, default=None, help="pkl motion file") |
|
parser.add_argument("--dir", type=str, default=None, help="pkl motion folder") |
|
parser.add_argument("--mode", type=str, default="sequence", help="render target: video, sequence, frame") |
|
parser.add_argument("--res", type=str, default="high") |
|
parser.add_argument("--denoising", type=bool, default=True) |
|
parser.add_argument("--oldrender", type=bool, default=True) |
|
parser.add_argument("--accelerator", type=str, default='gpu', help='accelerator device') |
|
parser.add_argument("--device", type=int, nargs='+', default=[0], help='gpu ids') |
|
parser.add_argument("--faces_path", type=str, default='./deps/smpl_models/smplh/smplh.faces') |
|
parser.add_argument("--always_on_floor", action="store_true", help='put all the body on the floor (not recommended)') |
|
parser.add_argument("--gt", type=str, default=False, help='green for gt, otherwise orange') |
|
parser.add_argument("--fps", type=int, default=20, help="the frame rate of the rendered video") |
|
parser.add_argument("--num", type=int, default=8, help="the number of frames rendered in 'sequence' mode") |
|
parser.add_argument("--exact_frame", type=float, default=0.5, help="the frame id selected under 'frame' mode ([0, 1])") |
|
cfg = parser.parse_args() |
|
return cfg |
|
|
|
|
|
def render_cli() -> None: |
|
cfg = parse_args() |
|
|
|
if cfg.pkl: |
|
paths = [cfg.pkl] |
|
elif cfg.dir: |
|
paths = [] |
|
file_list = natsort.natsorted(os.listdir(cfg.dir)) |
|
begin_id = random.randrange(0, len(file_list)) |
|
file_list = file_list[begin_id:] + file_list[:begin_id] |
|
|
|
for item in file_list: |
|
if item.endswith("_mesh.pkl"): |
|
paths.append(os.path.join(cfg.dir, item)) |
|
else: |
|
raise ValueError(f'{cfg.pkl} and {cfg.dir} are both None!') |
|
|
|
for path in paths: |
|
try: |
|
with open(path, 'rb') as f: |
|
pkl = pickle.load(f) |
|
data = pkl['vertices'] |
|
trajectory = pkl['hint'] |
|
|
|
except FileNotFoundError: |
|
print(f"{path} not found") |
|
continue |
|
|
|
render( |
|
data, |
|
trajectory, |
|
path, |
|
exact_frame=cfg.exact_frame, |
|
num=cfg.num, |
|
mode=cfg.mode, |
|
faces_path=cfg.faces_path, |
|
always_on_floor=cfg.always_on_floor, |
|
oldrender=cfg.oldrender, |
|
res=cfg.res, |
|
gt=cfg.gt, |
|
accelerator=cfg.accelerator, |
|
device=cfg.device, |
|
fps=cfg.fps) |
|
|
|
|
|
if __name__ == "__main__": |
|
render_cli() |
|
|
