mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-08-02 03:42:50 +08:00
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:
parent
9c161d1d04
commit
4bb3fff52b
@ -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)
|
||||
|
@ -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(
|
||||
|
@ -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(
|
||||
|
@ -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
|
||||
|
Loading…
x
Reference in New Issue
Block a user