Skip to content

GitLab

Unverified Commit de0c1039 authored by q.yao's avatar q.yao Committed by GitHub
Browse files

Fix bbox unit test (#1914)

parent 0e590362
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/test_ops/test_bbox.py
View file @ de0c1039
...@@ -8,7 +8,7 @@ from mmcv.utils import IS_CUDA_AVAILABLE, IS_MLU_AVAILABLE ...@@ -8,7 +8,7 @@ from mmcv.utils import IS_CUDA_AVAILABLE, IS_MLU_AVAILABLE
class TestBBox(object): class TestBBox(object):
def _test_bbox_overlaps(self, dtype=torch.float, device='cpu'): def _test_bbox_overlaps(self, device='cpu', dtype=torch.float):
from mmcv.ops import bbox_overlaps from mmcv.ops import bbox_overlaps
b1 = torch.tensor([[1.0, 1.0, 3.0, 4.0], [2.0, 2.0, 3.0, 4.0], b1 = torch.tensor([[1.0, 1.0, 3.0, 4.0], [2.0, 2.0, 3.0, 4.0],
[7.0, 7.0, 8.0, 8.0]]).to(device).type(dtype) [7.0, 7.0, 8.0, 8.0]]).to(device).type(dtype)
......
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