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

upgrade pyre version in fbcode/vision - batch 2

Browse Source 
Differential Revision: D42947615

fbshipit-source-id: 47b078fdf68567220e15993ab643f85771b0d340
This commit is contained in:
generatedunixname89002005307016 2023-02-01 20:28:20 -08:00 committed by Facebook GitHub Bot
parent 18c38ad600
commit d8471b26f2
14 changed files with 49 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pytorch3d/datasets/r2n2/r2n2.py
View File

@ -368,6 +368,8 @@ class R2N2(ShapeNetBase): # pragma: no cover
T = RT[3, :3]
return R, T
# pyre-fixme[14]: `render` overrides method defined in `ShapeNetBase`
# inconsistently.
def render(
self,
model_ids: Optional[List[str]] = None,

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

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

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

@ -315,6 +315,8 @@ class GenericModel(ImplicitronModelBase): # pyre-ignore: 13
self.log_loss_weights()
# pyre-fixme[14]: `forward` overrides method defined in `ImplicitronModelBase`
# inconsistently.
def forward(
self,
*, # force keyword-only arguments

4
pytorch3d/implicitron/models/implicit_function/scene_representation_networks.py
View File

@ -345,6 +345,8 @@ class SRNImplicitFunction(ImplicitFunctionBase, torch.nn.Module):
def raymarch_function_tweak_args(cls, type, args: DictConfig) -> None:
args.pop("latent_dim", None)
# pyre-fixme[14]: `forward` overrides method defined in `ImplicitFunctionBase`
# inconsistently.
def forward(
self,
*,
@ -404,6 +406,8 @@ class SRNHyperNetImplicitFunction(ImplicitFunctionBase, torch.nn.Module):
args.pop("latent_dim", None)
args.pop("latent_dim_hypernet", None)
# pyre-fixme[14]: `forward` overrides method defined in `ImplicitFunctionBase`
# inconsistently.
def forward(
self,
*,

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)
max_points: int = -1
# pyre-fixme[14]: `forward` overrides method defined in `ImplicitronModelBase`
# inconsistently.
def forward(
self,
*, # 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),
}[self.weight_function_type]
# pyre-fixme[14]: `forward` overrides method defined in `RaymarcherBase`
# inconsistently.
def forward(
self,
rays_densities: torch.Tensor,

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

@ -66,6 +66,8 @@ class SignedDistanceFunctionRenderer(BaseRenderer, torch.nn.Module): # pyre-ign
def requires_object_mask(self) -> bool:
return True
# pyre-fixme[14]: `forward` overrides method defined in `BaseRenderer`
# inconsistently.
def forward(
self,
ray_bundle: ImplicitronRayBundle,

4
pytorch3d/implicitron/tools/video_writer.py
View File

@ -86,6 +86,8 @@ class VideoWriter:
or a 2-tuple defining the size of the output image.
"""
# pyre-fixme[6]: For 1st argument expected `Union[PathLike[str], str]` but
# got `Optional[str]`.
outfile = os.path.join(self.cache_dir, self.regexp % self.frame_num)
if isinstance(frame, matplotlib.figure.Figure):
@ -130,6 +132,8 @@ class VideoWriter:
if self.frame_num == 0:
return ""
# pyre-fixme[6]: For 1st argument expected `Union[PathLike[str], str]` but
# got `Optional[str]`.
regexp = os.path.join(self.cache_dir, self.regexp)
if shutil.which(self.ffmpeg_bin) is None:

6
pytorch3d/io/obj_io.py
View File

@ -215,6 +215,8 @@ def load_obj(
"""
data_dir = "./"
if isinstance(f, (str, bytes, Path)):
# pyre-fixme[6]: For 1st argument expected `PathLike[Variable[AnyStr <:
# [str, bytes]]]` but got `Union[Path, bytes, str]`.
data_dir = os.path.dirname(f)
if path_manager is None:
path_manager = PathManager()
@ -296,6 +298,8 @@ class MeshObjFormat(MeshFormatInterpreter):
def __init__(self) -> None:
self.known_suffixes = (".obj",)
# pyre-fixme[14]: `read` overrides method defined in `MeshFormatInterpreter`
# inconsistently.
def read(
self,
path: PathOrStr,
@ -320,6 +324,8 @@ class MeshObjFormat(MeshFormatInterpreter):
)
return mesh
# pyre-fixme[14]: `save` overrides method defined in `MeshFormatInterpreter`
# inconsistently.
def save(
self,
data: Meshes,

2
pytorch3d/io/off_io.py
View File

@ -456,6 +456,8 @@ class MeshOffFormat(MeshFormatInterpreter):
)
return mesh
# pyre-fixme[14]: `save` overrides method defined in `MeshFormatInterpreter`
# inconsistently.
def save(
self,
data: Meshes,

4
pytorch3d/io/ply_io.py
View File

@ -1372,6 +1372,8 @@ class MeshPlyFormat(MeshFormatInterpreter):
)
return mesh
# pyre-fixme[14]: `save` overrides method defined in `MeshFormatInterpreter`
# inconsistently.
def save(
self,
data: Meshes,
@ -1452,6 +1454,8 @@ class PointcloudPlyFormat(PointcloudFormatInterpreter):
)
return pointcloud
# pyre-fixme[14]: `save` overrides method defined in
# `PointcloudFormatInterpreter` inconsistently.
def save(
self,
data: Pointclouds,

8
pytorch3d/renderer/cameras.py
View File

@ -695,6 +695,8 @@ class FoVPerspectiveCameras(CamerasBase):
)
return transform
# pyre-fixme[14]: `unproject_points` overrides method defined in `CamerasBase`
# inconsistently.
def unproject_points(
self,
xy_depth: torch.Tensor,
@ -938,6 +940,8 @@ class FoVOrthographicCameras(CamerasBase):
)
return transform
# pyre-fixme[14]: `unproject_points` overrides method defined in `CamerasBase`
# inconsistently.
def unproject_points(
self,
xy_depth: torch.Tensor,
@ -1145,6 +1149,8 @@ class PerspectiveCameras(CamerasBase):
)
return transform
# pyre-fixme[14]: `unproject_points` overrides method defined in `CamerasBase`
# inconsistently.
def unproject_points(
self,
xy_depth: torch.Tensor,
@ -1382,6 +1388,8 @@ class OrthographicCameras(CamerasBase):
)
return transform
# pyre-fixme[14]: `unproject_points` overrides method defined in `CamerasBase`
# inconsistently.
def unproject_points(
self,
xy_depth: torch.Tensor,

2
pytorch3d/renderer/fisheyecameras.py
View File

@ -347,6 +347,8 @@ class FishEyeCameras(CamerasBase):
point3d_est[..., :2] = theta.tan()[..., None] / xr_yrNorm[..., None] * xr_yr
return point3d_est
# pyre-fixme[14]: `unproject_points` overrides method defined in `CamerasBase`
# inconsistently.
def unproject_points(
self,
xy_depth: torch.Tensor,

8
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"]
return new_tex
# pyre-fixme[14]: `sample_textures` overrides method defined in `TexturesBase`
# inconsistently.
def sample_textures(self, fragments, **kwargs) -> torch.Tensor:
"""
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"]
return new_tex
# pyre-fixme[14]: `sample_textures` overrides method defined in `TexturesBase`
# inconsistently.
def sample_textures(self, fragments, **kwargs) -> torch.Tensor:
"""
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"]
return new_tex
# pyre-fixme[14]: `sample_textures` overrides method defined in `TexturesBase`
# inconsistently.
def sample_textures(self, fragments, faces_packed=None) -> torch.Tensor:
"""
Determine the color for each rasterized face. Interpolate the colors for
@ -1518,6 +1524,8 @@ class TexturesVertex(TexturesBase):
return self.__class__(sub_features)
# pyre-fixme[14]: `faces_verts_textures_packed` overrides method defined in
# `TexturesBase` inconsistently.
def faces_verts_textures_packed(self, faces_packed=None) -> torch.Tensor:
"""
Samples texture from each vertex and for each face in the mesh.