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)