mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-08-02 03:42:50 +08:00
upgrade pyre version in fbcode/vision
- batch 2
Differential Revision: D43044534 fbshipit-source-id: dc841b6704ccd562f5a40e7b2834e26063a9f7ae
This commit is contained in:
parent
d8471b26f2
commit
8c2b0b01f8
@ -368,8 +368,6 @@ class R2N2(ShapeNetBase): # pragma: no cover
|
|||||||
T = RT[3, :3]
|
T = RT[3, :3]
|
||||||
return R, T
|
return R, T
|
||||||
|
|
||||||
# pyre-fixme[14]: `render` overrides method defined in `ShapeNetBase`
|
|
||||||
# inconsistently.
|
|
||||||
def render(
|
def render(
|
||||||
self,
|
self,
|
||||||
model_ids: Optional[List[str]] = None,
|
model_ids: Optional[List[str]] = None,
|
||||||
|
@ -57,7 +57,6 @@ 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,
|
||||||
|
@ -315,8 +315,6 @@ class GenericModel(ImplicitronModelBase): # pyre-ignore: 13
|
|||||||
|
|
||||||
self.log_loss_weights()
|
self.log_loss_weights()
|
||||||
|
|
||||||
# pyre-fixme[14]: `forward` overrides method defined in `ImplicitronModelBase`
|
|
||||||
# inconsistently.
|
|
||||||
def forward(
|
def forward(
|
||||||
self,
|
self,
|
||||||
*, # force keyword-only arguments
|
*, # force keyword-only arguments
|
||||||
|
@ -345,8 +345,6 @@ class SRNImplicitFunction(ImplicitFunctionBase, torch.nn.Module):
|
|||||||
def raymarch_function_tweak_args(cls, type, args: DictConfig) -> None:
|
def raymarch_function_tweak_args(cls, type, args: DictConfig) -> None:
|
||||||
args.pop("latent_dim", None)
|
args.pop("latent_dim", None)
|
||||||
|
|
||||||
# pyre-fixme[14]: `forward` overrides method defined in `ImplicitFunctionBase`
|
|
||||||
# inconsistently.
|
|
||||||
def forward(
|
def forward(
|
||||||
self,
|
self,
|
||||||
*,
|
*,
|
||||||
@ -406,8 +404,6 @@ class SRNHyperNetImplicitFunction(ImplicitFunctionBase, torch.nn.Module):
|
|||||||
args.pop("latent_dim", None)
|
args.pop("latent_dim", None)
|
||||||
args.pop("latent_dim_hypernet", None)
|
args.pop("latent_dim_hypernet", None)
|
||||||
|
|
||||||
# pyre-fixme[14]: `forward` overrides method defined in `ImplicitFunctionBase`
|
|
||||||
# inconsistently.
|
|
||||||
def forward(
|
def forward(
|
||||||
self,
|
self,
|
||||||
*,
|
*,
|
||||||
|
@ -41,8 +41,6 @@ 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
|
||||||
|
@ -111,8 +111,6 @@ 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,
|
||||||
|
@ -66,8 +66,6 @@ class SignedDistanceFunctionRenderer(BaseRenderer, torch.nn.Module): # pyre-ign
|
|||||||
def requires_object_mask(self) -> bool:
|
def requires_object_mask(self) -> bool:
|
||||||
return True
|
return True
|
||||||
|
|
||||||
# pyre-fixme[14]: `forward` overrides method defined in `BaseRenderer`
|
|
||||||
# inconsistently.
|
|
||||||
def forward(
|
def forward(
|
||||||
self,
|
self,
|
||||||
ray_bundle: ImplicitronRayBundle,
|
ray_bundle: ImplicitronRayBundle,
|
||||||
|
@ -298,8 +298,6 @@ class MeshObjFormat(MeshFormatInterpreter):
|
|||||||
def __init__(self) -> None:
|
def __init__(self) -> None:
|
||||||
self.known_suffixes = (".obj",)
|
self.known_suffixes = (".obj",)
|
||||||
|
|
||||||
# pyre-fixme[14]: `read` overrides method defined in `MeshFormatInterpreter`
|
|
||||||
# inconsistently.
|
|
||||||
def read(
|
def read(
|
||||||
self,
|
self,
|
||||||
path: PathOrStr,
|
path: PathOrStr,
|
||||||
@ -324,8 +322,6 @@ class MeshObjFormat(MeshFormatInterpreter):
|
|||||||
)
|
)
|
||||||
return mesh
|
return mesh
|
||||||
|
|
||||||
# pyre-fixme[14]: `save` overrides method defined in `MeshFormatInterpreter`
|
|
||||||
# inconsistently.
|
|
||||||
def save(
|
def save(
|
||||||
self,
|
self,
|
||||||
data: Meshes,
|
data: Meshes,
|
||||||
|
@ -456,8 +456,6 @@ class MeshOffFormat(MeshFormatInterpreter):
|
|||||||
)
|
)
|
||||||
return mesh
|
return mesh
|
||||||
|
|
||||||
# pyre-fixme[14]: `save` overrides method defined in `MeshFormatInterpreter`
|
|
||||||
# inconsistently.
|
|
||||||
def save(
|
def save(
|
||||||
self,
|
self,
|
||||||
data: Meshes,
|
data: Meshes,
|
||||||
|
@ -1372,8 +1372,6 @@ class MeshPlyFormat(MeshFormatInterpreter):
|
|||||||
)
|
)
|
||||||
return mesh
|
return mesh
|
||||||
|
|
||||||
# pyre-fixme[14]: `save` overrides method defined in `MeshFormatInterpreter`
|
|
||||||
# inconsistently.
|
|
||||||
def save(
|
def save(
|
||||||
self,
|
self,
|
||||||
data: Meshes,
|
data: Meshes,
|
||||||
@ -1454,8 +1452,6 @@ class PointcloudPlyFormat(PointcloudFormatInterpreter):
|
|||||||
)
|
)
|
||||||
return pointcloud
|
return pointcloud
|
||||||
|
|
||||||
# pyre-fixme[14]: `save` overrides method defined in
|
|
||||||
# `PointcloudFormatInterpreter` inconsistently.
|
|
||||||
def save(
|
def save(
|
||||||
self,
|
self,
|
||||||
data: Pointclouds,
|
data: Pointclouds,
|
||||||
|
@ -695,8 +695,6 @@ class FoVPerspectiveCameras(CamerasBase):
|
|||||||
)
|
)
|
||||||
return transform
|
return transform
|
||||||
|
|
||||||
# pyre-fixme[14]: `unproject_points` overrides method defined in `CamerasBase`
|
|
||||||
# inconsistently.
|
|
||||||
def unproject_points(
|
def unproject_points(
|
||||||
self,
|
self,
|
||||||
xy_depth: torch.Tensor,
|
xy_depth: torch.Tensor,
|
||||||
@ -940,8 +938,6 @@ class FoVOrthographicCameras(CamerasBase):
|
|||||||
)
|
)
|
||||||
return transform
|
return transform
|
||||||
|
|
||||||
# pyre-fixme[14]: `unproject_points` overrides method defined in `CamerasBase`
|
|
||||||
# inconsistently.
|
|
||||||
def unproject_points(
|
def unproject_points(
|
||||||
self,
|
self,
|
||||||
xy_depth: torch.Tensor,
|
xy_depth: torch.Tensor,
|
||||||
@ -1149,8 +1145,6 @@ class PerspectiveCameras(CamerasBase):
|
|||||||
)
|
)
|
||||||
return transform
|
return transform
|
||||||
|
|
||||||
# pyre-fixme[14]: `unproject_points` overrides method defined in `CamerasBase`
|
|
||||||
# inconsistently.
|
|
||||||
def unproject_points(
|
def unproject_points(
|
||||||
self,
|
self,
|
||||||
xy_depth: torch.Tensor,
|
xy_depth: torch.Tensor,
|
||||||
@ -1388,8 +1382,6 @@ class OrthographicCameras(CamerasBase):
|
|||||||
)
|
)
|
||||||
return transform
|
return transform
|
||||||
|
|
||||||
# pyre-fixme[14]: `unproject_points` overrides method defined in `CamerasBase`
|
|
||||||
# inconsistently.
|
|
||||||
def unproject_points(
|
def unproject_points(
|
||||||
self,
|
self,
|
||||||
xy_depth: torch.Tensor,
|
xy_depth: torch.Tensor,
|
||||||
|
@ -347,8 +347,6 @@ class FishEyeCameras(CamerasBase):
|
|||||||
point3d_est[..., :2] = theta.tan()[..., None] / xr_yrNorm[..., None] * xr_yr
|
point3d_est[..., :2] = theta.tan()[..., None] / xr_yrNorm[..., None] * xr_yr
|
||||||
return point3d_est
|
return point3d_est
|
||||||
|
|
||||||
# pyre-fixme[14]: `unproject_points` overrides method defined in `CamerasBase`
|
|
||||||
# inconsistently.
|
|
||||||
def unproject_points(
|
def unproject_points(
|
||||||
self,
|
self,
|
||||||
xy_depth: torch.Tensor,
|
xy_depth: torch.Tensor,
|
||||||
|
@ -491,8 +491,6 @@ 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
|
||||||
@ -929,8 +927,6 @@ 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
|
||||||
@ -1454,8 +1450,6 @@ 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
|
||||||
@ -1524,8 +1518,6 @@ class TexturesVertex(TexturesBase):
|
|||||||
|
|
||||||
return self.__class__(sub_features)
|
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:
|
def faces_verts_textures_packed(self, faces_packed=None) -> torch.Tensor:
|
||||||
"""
|
"""
|
||||||
Samples texture from each vertex and for each face in the mesh.
|
Samples texture from each vertex and for each face in the mesh.
|
||||||
|
Loading…
x
Reference in New Issue
Block a user