423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-12-19 05:40:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

Bugfixes in render_flyaround

Browse Source 
Summary: Fixes bugs in render_flyaround

Reviewed By: bottler

Differential Revision: D39271932

fbshipit-source-id: 07e6c9ee07ba91feb437b725af0a8942fd98db0b
This commit is contained in:
David Novotny
2022-09-06 05:45:31 -07:00
committed by Facebook GitHub Bot
parent c79c954dea
commit f6d43eaa62
2 changed files with 38 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

59
tests/implicitron/test_model_visualize.py
View File

@@ -80,35 +80,36 @@ class TestModelVisualize(unittest.TestCase):
os.makedirs(output_dir, exist_ok=True)
render_flyaround(
train_dataset,
show_sequence_name,
model,
video_path,
n_flyaround_poses=40,
fps=20,
max_angle=2 * math.pi,
trajectory_type="circular_lsq_fit",
trajectory_scale=1.1,
scene_center=(0.0, 0.0, 0.0),
up=(0.0, 1.0, 0.0),
traj_offset=1.0,
n_source_views=1,
visdom_show_preds=visdom_show_preds,
visdom_environment="test_model_visalize",
visdom_server="http://127.0.0.1",
visdom_port=8097,
num_workers=10,
seed=None,
video_resize=None,
visualize_preds_keys=[
"images_render",
"depths_render",
"masks_render",
"_all_source_images",
],
output_video_frames_dir=video_path,
)
for output_video_frames_dir in [None, video_path]:
render_flyaround(
train_dataset,
show_sequence_name,
model,
video_path,
n_flyaround_poses=10,
fps=5,
max_angle=2 * math.pi,
trajectory_type="circular_lsq_fit",
trajectory_scale=1.1,
scene_center=(0.0, 0.0, 0.0),
up=(0.0, 1.0, 0.0),
traj_offset=1.0,
n_source_views=1,
visdom_show_preds=visdom_show_preds,
visdom_environment="test_model_visalize",
visdom_server="http://127.0.0.1",
visdom_port=8097,
num_workers=10,
seed=None,
video_resize=None,
visualize_preds_keys=[
"images_render",
"depths_render",
"masks_render",
"_all_source_images",
],
output_video_frames_dir=output_video_frames_dir,
)
class _PointcloudRenderingModel(torch.nn.Module):