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

More type annotations

Browse Source 
Summary: More type annotations: device, shaders, pluggable I/O, stats in NeRF project, cameras, textures, etc...

Reviewed By: nikhilaravi

Differential Revision: D29327396

fbshipit-source-id: cdf0ceaaa010e22423088752688c8dd81f1acc3c
This commit is contained in:
Patrick Labatut
2021-06-25 19:55:26 -07:00
committed by Facebook GitHub Bot
parent 542e2e7c07
commit f593bfd3c2
15 changed files with 196 additions and 148 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
pytorch3d/structures/volumes.py
View File

@@ -187,15 +187,15 @@ class Volumes:
"""
# handle densities
densities, grid_sizes = self._convert_densities_features_to_tensor(
densities_, grid_sizes = self._convert_densities_features_to_tensor(
densities, "densities"
)
# take device from densities
self.device = densities.device
self.device = densities_.device
# assign to the internal buffers
self._densities = densities
self._densities = densities_
self._grid_sizes = grid_sizes
# handle features
@@ -497,7 +497,6 @@ class Volumes:
)
def __len__(self) -> int:
# pyre-fixme[16]: `List` has no attribute `shape`.
return self._densities.shape[0]
def __getitem__(
@@ -547,8 +546,6 @@ class Volumes:
Returns:
**densities**: The tensor of volume densities.
"""
# pyre-fixme[7]: Expected `Tensor` but got `Union[List[torch.Tensor],
# torch.Tensor]`.
return self._densities
def densities_list(self) -> List[torch.Tensor]:
@@ -723,7 +720,6 @@ class Volumes:
return other
other.device = device_
# pyre-fixme[16]: `List` has no attribute `to`.
other._densities = self._densities.to(device_)
if self._features is not None:
# pyre-fixme[16]: `Optional` has no attribute `to`.