diff options
| author | Matthias Baumgartner <dev@igsor.net> | 2023-07-26 13:08:45 +0200 |
|---|---|---|
| committer | Matthias Baumgartner <dev@igsor.net> | 2023-07-26 13:08:45 +0200 |
| commit | 0bb413ac56ad035aab8b900430f413644e969fd2 (patch) | |
| tree | eedb5d3edd30a59d755565ea8c7684c485b3dca6 | |
| parent | bc6e80015844dfd4be44a877109e68798c26e79a (diff) | |
| download | bsie-0bb413ac56ad035aab8b900430f413644e969fd2.tar.gz bsie-0bb413ac56ad035aab8b900430f413644e969fd2.tar.bz2 bsie-0bb413ac56ad035aab8b900430f413644e969fd2.zip | |
bugfix
| -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, |