diff --git a/docs/examples/pulsar_basic_unified.py b/docs/examples/pulsar_basic_unified.py index dd44b128..f58c1e20 100755 --- a/docs/examples/pulsar_basic_unified.py +++ b/docs/examples/pulsar_basic_unified.py @@ -56,12 +56,12 @@ def cli(): focal_length=(5.0 * 2.0 / 2.0,), R=torch.eye(3, dtype=torch.float32, device=device)[None, ...], T=torch.zeros((1, 3), dtype=torch.float32, device=device), - image_size=((width, height),), + image_size=((height, width),), device=device, ) vert_rad = torch.rand(n_points, dtype=torch.float32, device=device) raster_settings = PointsRasterizationSettings( - image_size=(width, height), + image_size=(height, width), radius=vert_rad, ) rasterizer = PointsRasterizer(cameras=cameras, raster_settings=raster_settings) diff --git a/docs/examples/pulsar_cam_unified.py b/docs/examples/pulsar_cam_unified.py index 47affc4d..b0c451b0 100755 --- a/docs/examples/pulsar_cam_unified.py +++ b/docs/examples/pulsar_cam_unified.py @@ -123,11 +123,11 @@ class SceneModel(nn.Module): focal_length=self.focal_length, R=self.cam_rot[None, ...], T=self.cam_pos[None, ...], - image_size=((WIDTH, HEIGHT),), + image_size=((HEIGHT, WIDTH),), device=DEVICE, ) raster_settings = PointsRasterizationSettings( - image_size=(WIDTH, HEIGHT), + image_size=(HEIGHT, WIDTH), radius=self.vert_rad, ) rasterizer = PointsRasterizer( diff --git a/docs/examples/pulsar_optimization_unified.py b/docs/examples/pulsar_optimization_unified.py index 69517e1a..1cec4629 100755 --- a/docs/examples/pulsar_optimization_unified.py +++ b/docs/examples/pulsar_optimization_unified.py @@ -84,11 +84,11 @@ class SceneModel(nn.Module): focal_length=5.0, R=torch.eye(3, dtype=torch.float32, device=DEVICE)[None, ...], T=torch.zeros((1, 3), dtype=torch.float32, device=DEVICE), - image_size=((WIDTH, HEIGHT),), + image_size=((HEIGHT, WIDTH),), device=DEVICE, ) raster_settings = PointsRasterizationSettings( - image_size=(WIDTH, HEIGHT), + image_size=(HEIGHT, WIDTH), radius=self.vert_rad, ) rasterizer = PointsRasterizer( diff --git a/pytorch3d/renderer/points/pulsar/unified.py b/pytorch3d/renderer/points/pulsar/unified.py index 56ea3407..e003ae8f 100644 --- a/pytorch3d/renderer/points/pulsar/unified.py +++ b/pytorch3d/renderer/points/pulsar/unified.py @@ -80,7 +80,7 @@ class PulsarPointsRenderer(nn.Module): "by the pulsar backend." ) if isinstance(rasterizer.raster_settings.image_size, tuple): - width, height = rasterizer.raster_settings.image_size + height, width = rasterizer.raster_settings.image_size else: width = rasterizer.raster_settings.image_size height = rasterizer.raster_settings.image_size