remove nearest_neighbors

Summary: knn is more general and faster than the nearest_neighbor code, so remove the latter. Reviewed By: gkioxari Differential Revision: D20816424 fbshipit-source-id: 75d6c44d17180752d0c9859814bbdf7892558158
facebookresearch · Apr 16, 2020 · 3794f67 · 3794f67
1 parent 790eb8c
commit 3794f67
Show file tree

Hide file tree

Showing 6 changed files with 0 additions and 217 deletions.
diff --git a/pytorch3d/csrc/ext.cpp b/pytorch3d/csrc/ext.cpp
@@ -7,7 +7,6 @@
 #include "face_areas_normals/face_areas_normals.h"
 #include "gather_scatter/gather_scatter.h"
 #include "knn/knn.h"
-#include "nearest_neighbor_points/nearest_neighbor_points.h"
 #include "packed_to_padded_tensor/packed_to_padded_tensor.h"
 #include "point_mesh/point_mesh_edge.h"
 #include "point_mesh/point_mesh_face.h"
@@ -21,7 +20,6 @@ PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
   m.def("padded_to_packed", &PaddedToPacked);
   m.def("knn_points_idx", &KNearestNeighborIdx);
   m.def("knn_points_backward", &KNearestNeighborBackward);
-  m.def("nn_points_idx", &NearestNeighborIdx);
   m.def("gather_scatter", &gather_scatter);
   m.def("rasterize_points", &RasterizePoints);
   m.def("rasterize_points_backward", &RasterizePointsBackward);

diff --git a/pytorch3d/csrc/nearest_neighbor_points/nearest_neighbors_points_cpu.cpp b/pytorch3d/csrc/nearest_neighbor_points/nearest_neighbors_points_cpu.cpp
diff --git a/pytorch3d/ops/__init__.py b/pytorch3d/ops/__init__.py
@@ -5,7 +5,6 @@
 from .graph_conv import GraphConv
 from .knn import knn_gather, knn_points
 from .mesh_face_areas_normals import mesh_face_areas_normals
-from .nearest_neighbor_points import nn_points_idx
 from .packed_to_padded import packed_to_padded, padded_to_packed
 from .points_alignment import corresponding_points_alignment
 from .sample_points_from_meshes import sample_points_from_meshes

diff --git a/pytorch3d/ops/nearest_neighbor_points.py b/pytorch3d/ops/nearest_neighbor_points.py
diff --git a/tests/bm_nearest_neighbor_points.py b/tests/bm_nearest_neighbor_points.py
diff --git a/tests/test_nearest_neighbor_points.py b/tests/test_nearest_neighbor_points.py