mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-12-14 19:36:23 +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:
committed by
Facebook GitHub Bot
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(
|
||||
|
||||
Reference in New Issue
Block a user