upgrade pyre version in fbcode/vision - batch 2

Differential Revision: D55395614

fbshipit-source-id: 71677892b5d6f219f6df25b4efb51fb0f6b1441b
This commit is contained in:
generatedunixname89002005307016 2024-03-26 22:02:22 -07:00 committed by Facebook GitHub Bot
parent 38cf0dc1c5
commit b215776f2d
11 changed files with 39 additions and 0 deletions

View File

@ -6,5 +6,6 @@
# pyre-unsafe
# pyre-fixme[21]: Could not find module `pytorch3d.common.workaround.symeig3x3`.
from .symeig3x3 import symeig3x3
from .utils import _safe_det_3x3

View File

@ -8,8 +8,14 @@
from .chamfer import chamfer_distance
# pyre-fixme[21]: Could not find module `pytorch3d.loss.mesh_edge_loss`.
from .mesh_edge_loss import mesh_edge_loss
# pyre-fixme[21]: Could not find module `pytorch3d.loss.mesh_laplacian_smoothing`.
from .mesh_laplacian_smoothing import mesh_laplacian_smoothing
# pyre-fixme[21]: Could not find module `pytorch3d.loss.mesh_normal_consistency`.
from .mesh_normal_consistency import mesh_normal_consistency
from .point_mesh_distance import point_mesh_edge_distance, point_mesh_face_distance

View File

@ -6,16 +6,23 @@
# pyre-unsafe
# pyre-fixme[21]: Could not find module `pytorch3d.ops.ball_query`.
from .ball_query import ball_query
from .cameras_alignment import corresponding_cameras_alignment
# pyre-fixme[21]: Could not find module `pytorch3d.ops.cubify`.
from .cubify import cubify
from .graph_conv import GraphConv
from .interp_face_attrs import interpolate_face_attributes
from .iou_box3d import box3d_overlap
from .knn import knn_gather, knn_points
from .laplacian_matrices import cot_laplacian, laplacian, norm_laplacian
# pyre-fixme[21]: Could not find module `pytorch3d.ops.mesh_face_areas_normals`.
from .mesh_face_areas_normals import mesh_face_areas_normals
from .mesh_filtering import taubin_smoothing
# pyre-fixme[21]: Could not find module `pytorch3d.ops.packed_to_padded`.
from .packed_to_padded import packed_to_padded, padded_to_packed
from .perspective_n_points import efficient_pnp
from .points_alignment import corresponding_points_alignment, iterative_closest_point
@ -27,7 +34,11 @@ from .points_to_volumes import (
add_pointclouds_to_volumes,
add_points_features_to_volume_densities_features,
)
# pyre-fixme[21]: Could not find module `pytorch3d.ops.sample_farthest_points`.
from .sample_farthest_points import sample_farthest_points
# pyre-fixme[21]: Could not find module `pytorch3d.ops.sample_points_from_meshes`.
from .sample_points_from_meshes import sample_points_from_meshes
from .subdivide_meshes import SubdivideMeshes
from .utils import (
@ -37,6 +48,8 @@ from .utils import (
is_pointclouds,
wmean,
)
# pyre-fixme[21]: Could not find module `pytorch3d.ops.vert_align`.
from .vert_align import vert_align

View File

@ -15,7 +15,11 @@ import sys
from typing import Tuple, Union
import torch
# pyre-fixme[21]: Could not find module `pytorch3d.ops.mesh_face_areas_normals`.
from pytorch3d.ops.mesh_face_areas_normals import mesh_face_areas_normals
# pyre-fixme[21]: Could not find module `pytorch3d.ops.packed_to_padded`.
from pytorch3d.ops.packed_to_padded import packed_to_padded
from pytorch3d.renderer.mesh.rasterizer import Fragments as MeshFragments

View File

@ -12,6 +12,8 @@ from .clip import (
ClippedFaces,
convert_clipped_rasterization_to_original_faces,
)
# pyre-fixme[21]: Could not find module `pytorch3d.renderer.mesh.rasterize_meshes`.
from .rasterize_meshes import rasterize_meshes
from .rasterizer import MeshRasterizer, RasterizationSettings
from .renderer import MeshRenderer, MeshRendererWithFragments

View File

@ -13,6 +13,7 @@ import torch
import torch.nn as nn
from pytorch3d.renderer.cameras import try_get_projection_transform
# pyre-fixme[21]: Could not find module `pytorch3d.renderer.mesh.rasterize_meshes`.
from .rasterize_meshes import rasterize_meshes

View File

@ -8,6 +8,8 @@
from .compositor import AlphaCompositor, NormWeightedCompositor
from .pulsar.unified import PulsarPointsRenderer
# pyre-fixme[21]: Could not find module `pytorch3d.renderer.points.rasterize_points`.
from .rasterize_points import rasterize_points
from .rasterizer import PointsRasterizationSettings, PointsRasterizer
from .renderer import PointsRenderer

View File

@ -11,6 +11,8 @@ from typing import List, Optional, Tuple, Union
import numpy as np
import torch
from pytorch3d import _C
# pyre-fixme[21]: Could not find module `pytorch3d.renderer.mesh.rasterize_meshes`.
from pytorch3d.renderer.mesh.rasterize_meshes import pix_to_non_square_ndc
from ..utils import parse_image_size

View File

@ -15,6 +15,7 @@ import torch.nn as nn
from pytorch3d.renderer.cameras import try_get_projection_transform
from pytorch3d.structures import Pointclouds
# pyre-fixme[21]: Could not find module `pytorch3d.renderer.points.rasterize_points`.
from .rasterize_points import rasterize_points

View File

@ -853,6 +853,7 @@ class Meshes:
refresh: Set to True to force recomputation of face areas.
Default: False.
"""
# pyre-fixme[21]: Could not find module `pytorch3d.ops.mesh_face_areas_normals`.
from ..ops.mesh_face_areas_normals import mesh_face_areas_normals
if not (

View File

@ -12,8 +12,14 @@ from .camera_conversions import (
pulsar_from_cameras_projection,
pulsar_from_opencv_projection,
)
# pyre-fixme[21]: Could not find module `pytorch3d.utils.checkerboard`.
from .checkerboard import checkerboard
# pyre-fixme[21]: Could not find module `pytorch3d.utils.ico_sphere`.
from .ico_sphere import ico_sphere
# pyre-fixme[21]: Could not find module `pytorch3d.utils.torus`.
from .torus import torus