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

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