mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-08-03 04:12:48 +08:00
Correct shape for default grid_sizes
Summary: Small fix for omitting this argument. Reviewed By: nikhilaravi Differential Revision: D29548610 fbshipit-source-id: f25032fab3faa2f09006f5fcf8628138555f2f20
This commit is contained in:
parent
46cf1970ac
commit
b481cfbd01
@ -181,8 +181,11 @@ def add_points_features_to_volume_densities_features(
|
||||
|
||||
# init the volumetric grid sizes if uninitialized
|
||||
if grid_sizes is None:
|
||||
grid_sizes = torch.LongTensor(list(volume_densities.shape[2:])).to(
|
||||
volume_densities
|
||||
# grid sizes shape (minibatch, 3)
|
||||
grid_sizes = (
|
||||
torch.LongTensor(list(volume_densities.shape[2:]))
|
||||
.to(volume_densities)
|
||||
.expand(volume_densities.shape[0], 3)
|
||||
)
|
||||
|
||||
# flatten densities and features
|
||||
|
Loading…
x
Reference in New Issue
Block a user