🐞Fix the error if the device in masks_to_boxes is not both CPU and CUDA (open-edge-platform#1839)

* Check that the masks are CPU instead of CUDA

Signed-off-by: Danylo Boiko <[email protected]>

* Add a log to the changelog

Signed-off-by: Danylo Boiko <[email protected]>

---------

Signed-off-by: Danylo Boiko <[email protected]>
Signed-off-by: Bepitic <[email protected]>

Author: danylo-boiko

CHANGELOG.md

@@ -110,6 +110,7 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/).
 - 🐞Fix visualization by @ashwinvaidya17 in https://github.com/openvinotoolkit/anomalib/pull/1766
 - 🩹Minor fixes by @ashwinvaidya17 in https://github.com/openvinotoolkit/anomalib/pull/1788
 - ⏳ Restore Images by @ashwinvaidya17 in https://github.com/openvinotoolkit/anomalib/pull/1791
+- 🐞 Fix the error if the device in masks_to_boxes is not both CPU and CUDA by @danylo-boiko https://github.com/openvinotoolkit/anomalib/pull/1839
 
 ### New Contributors
 

src/anomalib/data/utils/boxes.py

@@ -31,10 +31,10 @@ def masks_to_boxes(
     if anomaly_maps is not None:
         anomaly_maps = anomaly_maps.view((-1,) + masks.shape[-2:])
 
-    if masks.is_cuda:
-        batch_comps = connected_components_gpu(masks).squeeze(1)
-    else:
+    if masks.is_cpu:
         batch_comps = connected_components_cpu(masks).squeeze(1)
+    else:
+        batch_comps = connected_components_gpu(masks).squeeze(1)
 
     batch_boxes = []
     batch_scores = []