Skip to content

Commit 72cad5b

Browse files
committed
simplify ImagePair conversion of CV-CUDA tensors
1 parent 78e892f commit 72cad5b

File tree

1 file changed

+4
-6
lines changed

1 file changed

+4
-6
lines changed

test/common_utils.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -299,12 +299,10 @@ def __init__(
299299
):
300300
if all(isinstance(input, PIL.Image.Image) for input in [actual, expected]):
301301
actual, expected = (to_image(input) for input in [actual, expected])
302-
elif CVCUDA_AVAILABLE and all(isinstance(input, _import_cvcuda().Tensor) for input in [actual, expected]):
303-
actual, expected = (cvcuda_to_tensor(input) for input in [actual, expected])
304-
elif CVCUDA_AVAILABLE and isinstance(actual, _import_cvcuda().Tensor) and isinstance(expected, PIL.Image.Image):
305-
actual = cvcuda_to_pil_compatible_tensor(actual)
306-
expected = to_image(expected)
307-
elif CVCUDA_AVAILABLE and isinstance(actual, _import_cvcuda().Tensor):
302+
303+
# handle check for CV-CUDA Tensors
304+
if CVCUDA_AVAILABLE and isinstance(actual, _import_cvcuda().Tensor):
305+
# Use the PIL compatible tensor, so we can always compare with PIL.Image.Image
308306
actual = cvcuda_to_pil_compatible_tensor(actual)
309307

310308
super().__init__(actual, expected, **other_parameters)

0 commit comments

Comments
 (0)