mirror of
				https://github.com/facebookresearch/pytorch3d.git
				synced 2025-11-04 18:02:14 +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]
 | 
			
		||||
        return R, T
 | 
			
		||||
 | 
			
		||||
    # pyre-fixme[14]: `render` overrides method defined in `ShapeNetBase`
 | 
			
		||||
    #  inconsistently.
 | 
			
		||||
    def render(
 | 
			
		||||
        self,
 | 
			
		||||
        model_ids: Optional[List[str]] = None,
 | 
			
		||||
 | 
			
		||||
@ -57,7 +57,6 @@ 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,
 | 
			
		||||
 | 
			
		||||
@ -315,8 +315,6 @@ 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
 | 
			
		||||
 | 
			
		||||
@ -345,8 +345,6 @@ 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,
 | 
			
		||||
        *,
 | 
			
		||||
@ -406,8 +404,6 @@ 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,
 | 
			
		||||
        *,
 | 
			
		||||
 | 
			
		||||
@ -41,8 +41,6 @@ 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
 | 
			
		||||
 | 
			
		||||
@ -111,8 +111,6 @@ 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,
 | 
			
		||||
 | 
			
		||||
@ -66,8 +66,6 @@ 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,
 | 
			
		||||
 | 
			
		||||
@ -298,8 +298,6 @@ 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,
 | 
			
		||||
@ -324,8 +322,6 @@ class MeshObjFormat(MeshFormatInterpreter):
 | 
			
		||||
        )
 | 
			
		||||
        return mesh
 | 
			
		||||
 | 
			
		||||
    # pyre-fixme[14]: `save` overrides method defined in `MeshFormatInterpreter`
 | 
			
		||||
    #  inconsistently.
 | 
			
		||||
    def save(
 | 
			
		||||
        self,
 | 
			
		||||
        data: Meshes,
 | 
			
		||||
 | 
			
		||||
@ -456,8 +456,6 @@ class MeshOffFormat(MeshFormatInterpreter):
 | 
			
		||||
        )
 | 
			
		||||
        return mesh
 | 
			
		||||
 | 
			
		||||
    # pyre-fixme[14]: `save` overrides method defined in `MeshFormatInterpreter`
 | 
			
		||||
    #  inconsistently.
 | 
			
		||||
    def save(
 | 
			
		||||
        self,
 | 
			
		||||
        data: Meshes,
 | 
			
		||||
 | 
			
		||||
@ -1372,8 +1372,6 @@ class MeshPlyFormat(MeshFormatInterpreter):
 | 
			
		||||
        )
 | 
			
		||||
        return mesh
 | 
			
		||||
 | 
			
		||||
    # pyre-fixme[14]: `save` overrides method defined in `MeshFormatInterpreter`
 | 
			
		||||
    #  inconsistently.
 | 
			
		||||
    def save(
 | 
			
		||||
        self,
 | 
			
		||||
        data: Meshes,
 | 
			
		||||
@ -1454,8 +1452,6 @@ class PointcloudPlyFormat(PointcloudFormatInterpreter):
 | 
			
		||||
        )
 | 
			
		||||
        return pointcloud
 | 
			
		||||
 | 
			
		||||
    # pyre-fixme[14]: `save` overrides method defined in
 | 
			
		||||
    #  `PointcloudFormatInterpreter` inconsistently.
 | 
			
		||||
    def save(
 | 
			
		||||
        self,
 | 
			
		||||
        data: Pointclouds,
 | 
			
		||||
 | 
			
		||||
@ -695,8 +695,6 @@ class FoVPerspectiveCameras(CamerasBase):
 | 
			
		||||
        )
 | 
			
		||||
        return transform
 | 
			
		||||
 | 
			
		||||
    # pyre-fixme[14]: `unproject_points` overrides method defined in `CamerasBase`
 | 
			
		||||
    #  inconsistently.
 | 
			
		||||
    def unproject_points(
 | 
			
		||||
        self,
 | 
			
		||||
        xy_depth: torch.Tensor,
 | 
			
		||||
@ -940,8 +938,6 @@ class FoVOrthographicCameras(CamerasBase):
 | 
			
		||||
        )
 | 
			
		||||
        return transform
 | 
			
		||||
 | 
			
		||||
    # pyre-fixme[14]: `unproject_points` overrides method defined in `CamerasBase`
 | 
			
		||||
    #  inconsistently.
 | 
			
		||||
    def unproject_points(
 | 
			
		||||
        self,
 | 
			
		||||
        xy_depth: torch.Tensor,
 | 
			
		||||
@ -1149,8 +1145,6 @@ class PerspectiveCameras(CamerasBase):
 | 
			
		||||
        )
 | 
			
		||||
        return transform
 | 
			
		||||
 | 
			
		||||
    # pyre-fixme[14]: `unproject_points` overrides method defined in `CamerasBase`
 | 
			
		||||
    #  inconsistently.
 | 
			
		||||
    def unproject_points(
 | 
			
		||||
        self,
 | 
			
		||||
        xy_depth: torch.Tensor,
 | 
			
		||||
@ -1388,8 +1382,6 @@ class OrthographicCameras(CamerasBase):
 | 
			
		||||
        )
 | 
			
		||||
        return transform
 | 
			
		||||
 | 
			
		||||
    # pyre-fixme[14]: `unproject_points` overrides method defined in `CamerasBase`
 | 
			
		||||
    #  inconsistently.
 | 
			
		||||
    def unproject_points(
 | 
			
		||||
        self,
 | 
			
		||||
        xy_depth: torch.Tensor,
 | 
			
		||||
 | 
			
		||||
@ -347,8 +347,6 @@ 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,
 | 
			
		||||
 | 
			
		||||
@ -491,8 +491,6 @@ 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
 | 
			
		||||
@ -929,8 +927,6 @@ 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
 | 
			
		||||
@ -1454,8 +1450,6 @@ 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
 | 
			
		||||
@ -1524,8 +1518,6 @@ 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.
 | 
			
		||||
 | 
			
		||||
		Loading…
	
	
			
			x
			
			
		
	
		Reference in New Issue
	
	Block a user