fix hungarian matcher

openvinotoolkit · Dec 19, 2024 · b8c912d · b8c912d
1 parent a409186
commit b8c912d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/otx/algo/common/utils/assigners/hungarian_matcher.py b/src/otx/algo/common/utils/assigners/hungarian_matcher.py
@@ -279,7 +279,7 @@ def batch_preparation(
                 "pred_boxes": outputs["pred_boxes"][i],
                 "pred_masks": outputs["pred_masks"][i] if "pred_masks" in outputs else None,
                 "target_boxes": targets[i]["boxes"],
-                "target_labels": targets[i]["labels"],
+                "target_labels": targets[i]["labels"].long(),
                 "target_mask": targets[i]["masks"] if "masks" in targets[i] else None,
             }
             for i in range(batch_size)