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

upgrade pyre version in fbcode/vision - batch 2

Browse Source 
Differential Revision: D44881859

fbshipit-source-id: 4ed410724a14d580f811c1288f51a71ce8fb0c9a
This commit is contained in:
generatedunixname89002005307016 2023-04-11 17:15:12 -07:00 committed by Facebook GitHub Bot
parent e245560abb
commit 355d6332cb
4 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
pytorch3d/implicitron/evaluation/evaluator.py
View File

@ -57,6 +57,7 @@ class ImplicitronEvaluator(EvaluatorBase):
def __post_init__(self): def __post_init__(self):
run_auto_creation(self) run_auto_creation(self)
# pyre-fixme[14]: `run` overrides method defined in `EvaluatorBase` inconsistently.
def run( def run(
self, self,
model: ImplicitronModelBase, model: ImplicitronModelBase,

2
pytorch3d/implicitron/models/model_dbir.py
View File

@ -41,6 +41,8 @@ class ModelDBIR(ImplicitronModelBase):
bg_color: Tuple[float, float, float] = (0.0, 0.0, 0.0) bg_color: Tuple[float, float, float] = (0.0, 0.0, 0.0)
max_points: int = -1 max_points: int = -1
# pyre-fixme[14]: `forward` overrides method defined in `ImplicitronModelBase`
# inconsistently.
def forward( def forward(
self, self,
*, # force keyword-only arguments *, # force keyword-only arguments

2
pytorch3d/implicitron/models/renderer/raymarcher.py
View File

@ -111,6 +111,8 @@ class AccumulativeRaymarcherBase(RaymarcherBase, torch.nn.Module):
"minimum": lambda curr, acc: torch.minimum(curr, acc), "minimum": lambda curr, acc: torch.minimum(curr, acc),
}[self.weight_function_type] }[self.weight_function_type]
# pyre-fixme[14]: `forward` overrides method defined in `RaymarcherBase`
# inconsistently.
def forward( def forward(
self, self,
rays_densities: torch.Tensor, rays_densities: torch.Tensor,

6
pytorch3d/renderer/mesh/textures.py
View File

@ -491,6 +491,8 @@ class TexturesAtlas(TexturesBase):
new_tex._num_faces_per_mesh = new_props["_num_faces_per_mesh"] new_tex._num_faces_per_mesh = new_props["_num_faces_per_mesh"]
return new_tex return new_tex
# pyre-fixme[14]: `sample_textures` overrides method defined in `TexturesBase`
# inconsistently.
def sample_textures(self, fragments, **kwargs) -> torch.Tensor: def sample_textures(self, fragments, **kwargs) -> torch.Tensor:
""" """
This is similar to a nearest neighbor sampling and involves a This is similar to a nearest neighbor sampling and involves a
@ -927,6 +929,8 @@ class TexturesUV(TexturesBase):
new_tex._num_faces_per_mesh = new_props["_num_faces_per_mesh"] new_tex._num_faces_per_mesh = new_props["_num_faces_per_mesh"]
return new_tex return new_tex
# pyre-fixme[14]: `sample_textures` overrides method defined in `TexturesBase`
# inconsistently.
def sample_textures(self, fragments, **kwargs) -> torch.Tensor: def sample_textures(self, fragments, **kwargs) -> torch.Tensor:
""" """
Interpolate a 2D texture map using uv vertex texture coordinates for each Interpolate a 2D texture map using uv vertex texture coordinates for each
@ -1450,6 +1454,8 @@ class TexturesVertex(TexturesBase):
new_tex._num_verts_per_mesh = new_props["_num_verts_per_mesh"] new_tex._num_verts_per_mesh = new_props["_num_verts_per_mesh"]
return new_tex return new_tex
# pyre-fixme[14]: `sample_textures` overrides method defined in `TexturesBase`
# inconsistently.
def sample_textures(self, fragments, faces_packed=None) -> torch.Tensor: def sample_textures(self, fragments, faces_packed=None) -> torch.Tensor:
""" """
Determine the color for each rasterized face. Interpolate the colors for Determine the color for each rasterized face. Interpolate the colors for