423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2026-02-25 23:56:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

fbcode/vision/fair/pytorch3d/pytorch3d/ops/points_alignment.py

Browse Source 
Reviewed By: sgrigory

Differential Revision: D93712744

fbshipit-source-id: 660560cdef9ff1d2173ae06de54df31766ee537f
This commit is contained in:
generatedunixname1417043136753450
2026-02-23 12:28:37 -08:00
committed by meta-codesync[bot]
parent d631b56fba
commit 0e435c297c
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
pytorch3d/ops/points_alignment.py
View File

@@ -182,8 +182,7 @@ def iterative_closest_point(
t_history.append(SimilarityTransform(R, T, s))
# compute the root mean squared error
# pyre-fixme[58]: `**` is not supported for operand types `Tensor` and `int`.
Xt_sq_diff = ((Xt - Xt_nn_points) ** 2).sum(2)
Xt_sq_diff = torch.square(Xt - Xt_nn_points).sum(2)
rmse = oputil.wmean(Xt_sq_diff[:, :, None], mask_X).sqrt()[:, 0, 0]
# compute the relative rmse