forked from facebookresearch/KeypointNeRF
-
Notifications
You must be signed in to change notification settings - Fork 0
/
render_dynamic.py
37 lines (29 loc) · 1.25 KB
/
render_dynamic.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
# Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
# This source code is licensed under the license found in the
# LICENSE file in the root directory of this source tree.
import torch
from src import config
from src.model import KeypointNeRFLightningModule
from src.zju_dataset import ZJUDataset
if __name__ == "__main__":
torch.set_default_dtype(torch.float32)
torch.autograd.set_detect_anomaly(True)
# load configuration
parser = config.create_parser()
args = parser.parse_args(None)
cfg = config.load_cfg(args.config)
cfg['dataset']['data_root'] = args.data_root
if args.out_dir is not None:
cfg['out_dir'] = args.out_dir
cfg['expname'] = cfg.get('expname', 'default')
# create data loader
test_dataset = ZJUDataset.from_config(cfg['dataset'], 'test_visualize', cfg)
data_loader = torch.utils.data.DataLoader(
test_dataset, shuffle=False, pin_memory=True,
num_workers=cfg['training'].get('val_num_workers', 0),
collate_fn=KeypointNeRFLightningModule.collate_fn)
# create model
renderer = KeypointNeRFLightningModule.from_config(cfg, cfg.get('method', None)).cuda()
renderer.load_ckpt(args.model_ckpt)
renderer.render_video_zju(data_loader)