Skip to content
Snippets Groups Projects
Unverified Commit a56185ab authored by Tai-Wang's avatar Tai-Wang Committed by GitHub
Browse files

[Fix] Fix nuscenes_mono_dataset unit test due to hack fix (#795)

* Fix nuscenes_mono_dataset unit test due to hack fix

* Remove unused codes
parent 77964873
Branches
No related tags found
No related merge requests found
......@@ -135,11 +135,8 @@ def test_format_results():
expected_token = 'e93e98b63d3b40209056d129dc53ceee'
expected_trans = torch.tensor(
[1018.753821915645, 605.190386124652, 0.7266818822266328])
expected_size = torch.tensor([1.6380000114440918, 4.25, 1.440000057220459])
expected_rotation = torch.tensor([
-0.9924980733795628, -0.013604682549109839, 0.01027292674776989,
-0.12106590736714223
])
expected_size = torch.tensor([1.440000057220459, 1.6380000114440918, 4.25])
expected_rotation = torch.tensor([-0.5717, -0.0014, 0.0170, -0.8203])
expected_detname = 'car'
expected_attr = 'vehicle.moving'
......@@ -148,7 +145,7 @@ def test_format_results():
torch.tensor(det['translation']), expected_trans, 1e-5)
assert torch.allclose(torch.tensor(det['size']), expected_size, 1e-5)
assert torch.allclose(
torch.tensor(det['rotation']), expected_rotation, 1e-5)
torch.tensor(det['rotation']), expected_rotation, atol=1e-4)
assert det['detection_name'] == expected_detname
assert det['attribute_name'] == expected_attr
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment