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

KNN return order documentation

Browse Source 
Summary: Fix documentation of KNN, issue #180

Reviewed By: gkioxari

Differential Revision: D21384761

fbshipit-source-id: 2b36ee496f2060d17827d2fd66c490cdfa766866
This commit is contained in:
Jeremy Reizenstein
2020-05-04 13:35:49 -07:00
committed by Facebook GitHub Bot
parent 0eca74fa5f
commit 06ad1fb6c4
3 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pytorch3d/ops/points_alignment.py
View File

@@ -158,7 +158,7 @@ def iterative_closest_point(
for iteration in range(max_iterations):
Xt_nn_points = knn_points(
Xt, Yt, lengths1=num_points_X, lengths2=num_points_Y, K=1, return_nn=True
)[2][:, :, 0, :]
).knn[:, :, 0, :]
# get the alignment of the nearest neighbors from Yt with Xt_init
R, T, s = corresponding_points_alignment(