423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-08-02 11:52:50 +08:00
Code Issues Packages Projects Releases Wiki Activity
pytorch3d/tests
History
Georgia Gkioxari b2b0c5a442 knn autograd 
Summary:
Adds knn backward to return `grad_pts1` and `grad_pts2`. Adds `knn_gather` to return the nearest neighbors in pts2.

The BM tests include backward pass and are ran on an M40.
```
Benchmark                               Avg Time(μs)      Peak Time(μs) Iterations
--------------------------------------------------------------------------------
KNN_SQUARE_32_256_128_3_24_cpu              39558           43485             13
KNN_SQUARE_32_256_128_3_24_cuda:0            1080            1404            463
KNN_SQUARE_32_256_512_3_24_cpu              81950           85781              7
KNN_SQUARE_32_256_512_3_24_cuda:0            1519            1641            330
--------------------------------------------------------------------------------

Benchmark                               Avg Time(μs)      Peak Time(μs) Iterations
--------------------------------------------------------------------------------
KNN_RAGGED_32_256_128_3_24_cpu              13798           14650             37
KNN_RAGGED_32_256_128_3_24_cuda:0            1576            1713            318
KNN_RAGGED_32_256_512_3_24_cpu              31255           32210             16
KNN_RAGGED_32_256_512_3_24_cuda:0            2024            2162            248
--------------------------------------------------------------------------------
```

Reviewed By: jcjohnson

Differential Revision: D20945556

fbshipit-source-id: a16f616029c6b5f8c2afceb5f2bc12c5c20d2f3c
2020-04-14 17:22:56 -07:00
..
data
Rendering texturing fixes
2020-03-17 08:58:40 -07:00
__init__.py
Initial commit
2020-01-23 11:53:46 -08:00
bm_blending.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
bm_chamfer.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
bm_cubify.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
bm_face_areas_normals.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
bm_graph_conv.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
bm_knn.py
knn autograd
2020-04-14 17:22:56 -07:00
bm_main.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
bm_mesh_edge_loss.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
bm_mesh_io.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
bm_mesh_laplacian_smoothing.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
bm_mesh_normal_consistency.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
bm_meshes.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
bm_nearest_neighbor_points.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
bm_packed_to_padded.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
bm_point_mesh_distance.py
point mesh distances
2020-04-11 00:21:24 -07:00
bm_pointclouds.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
bm_points_alignment.py
fix recent lint
2020-04-06 06:41:00 -07:00
bm_rasterize_meshes.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
bm_rasterize_points.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
bm_sample_points_from_meshes.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
bm_so3.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
bm_subdivide_meshes.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
bm_vert_align.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
common_testing.py
fix recent lint
2020-04-06 06:41:00 -07:00
test_blending.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
test_build.py
Allow conda's generated files.
2020-04-06 10:03:57 -07:00
test_cameras.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
test_chamfer.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
test_compositing.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
test_cubify.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
test_face_areas_normals.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
test_graph_conv.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
test_knn.py
knn autograd
2020-04-14 17:22:56 -07:00
test_lighting.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
test_materials.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
test_mesh_edge_loss.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
test_mesh_laplacian_smoothing.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
test_mesh_normal_consistency.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
test_mesh_rendering_utils.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
test_meshes.py
point mesh distances
2020-04-11 00:21:24 -07:00
test_nearest_neighbor_points.py
fix recent lint
2020-03-30 06:17:27 -07:00
test_obj_io.py
join_meshes_as_batch
2020-03-30 11:27:41 -07:00
test_ops_utils.py
fix recent lint
2020-04-06 06:41:00 -07:00
test_packed_to_padded.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
test_ply_io.py
Fix saving / loading empty PLY meshes
2020-04-01 04:51:20 -07:00
test_point_mesh_distance.py
point mesh distances
2020-04-11 00:21:24 -07:00
test_pointclouds.py
point mesh distances
2020-04-11 00:21:24 -07:00
test_points_alignment.py
fix recent lint
2020-04-06 06:41:00 -07:00
test_rasterize_meshes.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
test_rasterize_points.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
test_rasterizer.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
test_rendering_meshes.py
point mesh distances
2020-04-11 00:21:24 -07:00
test_rendering_utils.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
test_rotation_conversions.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
test_sample_points_from_meshes.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
test_so3.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
test_struct_utils.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
test_subdivide_meshes.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
test_texturing.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
test_transforms.py
Initialization of Transform3D with a custom matrix.
2020-04-05 14:44:27 -07:00
test_vert_align.py
Address black + isort fbsource linter warnings
2020-03-29 14:51:02 -07:00
weights.pt
Initial commit
2020-01-23 11:53:46 -08:00