pulsar image size bug fix

Summary: Small change to swap how height/width are inferred from the image_size setting.

Reviewed By: gkioxari

Differential Revision: D26648340

fbshipit-source-id: 2c657a115c96cadf3ac63be87b0e1bfba10c9315
This commit is contained in:
Nikhila Ravi 2021-02-26 09:07:37 -08:00 committed by Facebook GitHub Bot
parent 9c161d1d04
commit 4bb3fff52b
4 changed files with 7 additions and 7 deletions

View File

@ -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)

View File

@ -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(

View File

@ -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(

View File

@ -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