423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-08-23 14:22:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

Correct shape for default grid_sizes

Browse Source 
Summary: Small fix for omitting this argument.

Reviewed By: nikhilaravi

Differential Revision: D29548610

fbshipit-source-id: f25032fab3faa2f09006f5fcf8628138555f2f20
This commit is contained in:
Jeremy Reizenstein 2021-08-17 05:57:43 -07:00 committed by Facebook GitHub Bot
parent 46cf1970ac
commit b481cfbd01
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
pytorch3d/ops/points_to_volumes.py
View File

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