diff --git a/models/tensorBase.py b/models/tensorBase.py index 0eb863d..7aced71 100644 --- a/models/tensorBase.py +++ b/models/tensorBase.py @@ -307,7 +307,7 @@ def getDenseAlpha(self,gridSize=None): torch.linspace(0, 1, gridSize[0]), torch.linspace(0, 1, gridSize[1]), torch.linspace(0, 1, gridSize[2]), - ), -1).to(self.device) + ), -1, indexing='ij').to(self.device) dense_xyz = self.aabb[0] * (1-samples) + self.aabb[1] * samples # dense_xyz = dense_xyz