Function main

motion_rep/motion_checker.py:548–581

()

def main():
    parser = argparse.ArgumentParser(description="Visualize Depth Maps from Motion File")
    parser.add_argument("--motion_file", type=str, required=True, help="Path to .pt motion file")
    parser.add_argument("--output_dir", type=str, required=True, help="Output directory for visualizations")
    parser.add_argument("--video_file", type=str, default=None, help="Optional RGB video to overlay depth (overrides H/W/fps)")
    parser.add_argument("--smpl_type", type=str, default="smplx", choices=["smplx", "smplh"], help="Body model type")
    parser.add_argument("--smpl_model_path", type=str, default=None, help="Override body model directory")
    parser.add_argument("--batch_size", type=int, default=24, help="Batch size for rendering frames")
    parser.add_argument("--H", type=int, default=1080, help="Image height (ignored if video_file provided)")
    parser.add_argument("--W", type=int, default=1920, help="Image width (ignored if video_file provided)")
    parser.add_argument("--fps", type=int, default=20, help="Frames per second (ignored if video_file provided)")
    parser.add_argument("--recover_from_velocity", "-rfv", action="store_true", help="Recover params from velocity")
    parser.add_argument("--zero_trans", "-zt", action="store_true", help="Set translation to zero")
    parser.add_argument("--device", type=str, default="cuda:0", help="Device to run model on")
    parser.add_argument("--verbose", "-v", action="store_true", help="Verbose mode")
    args = parser.parse_args()

    device = args.device
    motion_vis(
        args.motion_file,
        args.output_dir,
        args.batch_size,
        args.H,
        args.W,
        args.fps,
        None,
        args.recover_from_velocity,
        device=device,
        verbose=args.verbose,
        zero_trans=args.zero_trans,
        video_file=args.video_file,
        smpl_model_path=args.smpl_model_path,
        smpl_type=args.smpl_type,
    )


if __name__ == "__main__":