diff --git a/pytorch3d/ops/subdivide_meshes.py b/pytorch3d/ops/subdivide_meshes.py index 6a2386e8..417a1447 100644 --- a/pytorch3d/ops/subdivide_meshes.py +++ b/pytorch3d/ops/subdivide_meshes.py @@ -103,8 +103,9 @@ class SubdivideMeshes(nn.Module): verts_packed = meshes.verts_packed() with torch.no_grad(): faces_packed = meshes.faces_packed() - faces_packed_to_edges_packed = meshes.faces_packed_to_edges_packed() - faces_packed_to_edges_packed += verts_packed.shape[0] + faces_packed_to_edges_packed = ( + meshes.faces_packed_to_edges_packed() + verts_packed.shape[0] + ) f0 = torch.stack( [ diff --git a/pytorch3d/structures/meshes.py b/pytorch3d/structures/meshes.py index a911223c..e1181b0b 100644 --- a/pytorch3d/structures/meshes.py +++ b/pytorch3d/structures/meshes.py @@ -1035,9 +1035,7 @@ class Meshes(object): self._edges_packed = torch.stack([u // V, u % V], dim=1) self._edges_packed_to_mesh_idx = edge_to_mesh[unique_idx] - face_to_edge = torch.arange(3 * F).view(3, F).t() - face_to_edge = inverse_idxs[face_to_edge] - self._faces_packed_to_edges_packed = face_to_edge + self._faces_packed_to_edges_packed = inverse_idxs.reshape(3, F).t() # Compute number of edges per mesh num_edges_per_mesh = torch.zeros(self._N, dtype=torch.int32, device=self.device)