From 0bb413ac56ad035aab8b900430f413644e969fd2 Mon Sep 17 00:00:00 2001 From: Matthias Baumgartner Date: Wed, 26 Jul 2023 13:08:45 +0200 Subject: bugfix --- bsie/extractor/image/face/identify.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/bsie/extractor/image/face/identify.py b/bsie/extractor/image/face/identify.py index dee935f..2a7efa2 100644 --- a/bsie/extractor/image/face/identify.py +++ b/bsie/extractor/image/face/identify.py @@ -139,7 +139,7 @@ class FaceIdentify(base.Extractor): best = dist.argmin(dim=1) # idx of lowest distance, per row labels = self._targets[best] # label (int) of nearest neighbour acc = dist[range(len(best)), best] < self._thres # check if distance is below threshold - return [lbl.item() if cnd is True else self._restidx for cnd, lbl in zip(acc, labels)] + return [lbl.item() if cnd == True else self._restidx for cnd, lbl in zip(acc, labels)] # pylint: disable=singleton-comparison def extract( self, -- cgit v1.2.3