diff options
Diffstat (limited to 'bsie/extractor/image/face/identify.py')
| -rw-r--r-- | bsie/extractor/image/face/identify.py | 2 |
1 files changed, 1 insertions, 1 deletions
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, |