Skip to content

GitLab

"docs/git@developer.sourcefind.cn:OpenDAS/mmdetection3d.git" did not exist on "31f1145d0edec688ea48d31ac76d51e60eb4bda4"
Unverified Commit 6356cbdc authored by encore-zhou's avatar encore-zhou Committed by GitHub
Browse files

[Fix]: add missing test data for pointnet modules (#85) 

* add test data

* update fps test data

* update fps test data
parent 4040a074
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/test_pointnet_ops.py
View file @ 6356cbdc
...@@ -340,5 +340,5 @@ def test_fps_with_dist(): ...@@ -340,5 +340,5 @@ def test_fps_with_dist():
expected_idx = torch.from_numpy(fps_idx).cuda() expected_idx = torch.from_numpy(fps_idx).cuda()
features_for_fps_distance = torch.from_numpy( features_for_fps_distance = torch.from_numpy(
features_for_fps_distance).cuda() features_for_fps_distance).cuda()
idx = furthest_point_sample_with_dist(features_for_fps_distance, 32) idx = furthest_point_sample_with_dist(features_for_fps_distance, 16)
assert torch.all(idx == expected_idx) assert torch.all(idx == expected_idx)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment