423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2026-03-04 03:05:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix recent lint

Browse Source 
Summary: lint clean again

Reviewed By: patricklabatut

Differential Revision: D20868775

fbshipit-source-id: ade4301c1012c5c6943186432465215701d635a9
This commit is contained in:
Jeremy Reizenstein
2020-04-06 06:38:50 -07:00
committed by Facebook GitHub Bot
parent 90dc7a0856
commit b87058c62a
6 changed files with 29 additions and 60 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
pytorch3d/ops/points_alignment.py
View File

@@ -1,12 +1,12 @@
# Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.
import warnings
from typing import List, Optional, Tuple, Union
import torch
from typing import List, Tuple, Union
from pytorch3d.structures.pointclouds import Pointclouds
from pytorch3d.structures import utils as strutil
import torch
from pytorch3d.ops import utils as oputil
from pytorch3d.structures import utils as strutil
from pytorch3d.structures.pointclouds import Pointclouds
def corresponding_points_alignment(
@@ -77,9 +77,7 @@ def corresponding_points_alignment(
weights = strutil.list_to_padded(weights)[..., 0]
if Xt.shape[:2] != weights.shape:
raise ValueError(
"weights should have the same first two dimensions as X."
)
raise ValueError("weights should have the same first two dimensions as X.")
b, n, dim = Xt.shape
@@ -120,9 +118,7 @@ def corresponding_points_alignment(
U, S, V = torch.svd(XYcov)
# identity matrix used for fixing reflections
E = torch.eye(dim, dtype=XYcov.dtype, device=XYcov.device)[None].repeat(
b, 1, 1
)
E = torch.eye(dim, dtype=XYcov.dtype, device=XYcov.device)[None].repeat(b, 1, 1)
if not allow_reflection:
# reflection test: