diff options
| author | Matthias Baumgartner <dev@igsor.net> | 2023-07-26 12:48:54 +0200 |
|---|---|---|
| committer | Matthias Baumgartner <dev@igsor.net> | 2023-07-26 12:48:54 +0200 |
| commit | d6a5c0f596a212f0e1d4e4b351b5b0e6857d74f7 (patch) | |
| tree | 645c6fe6b120bb4759b7ac07b4799ffec3bfb4bf /bsie/extractor/image/face/identify.py | |
| parent | b1ee4452c0e4b820efe69e428e7eaa54cf87ae16 (diff) | |
| download | bsie-d6a5c0f596a212f0e1d4e4b351b5b0e6857d74f7.tar.gz bsie-d6a5c0f596a212f0e1d4e4b351b5b0e6857d74f7.tar.bz2 bsie-d6a5c0f596a212f0e1d4e4b351b5b0e6857d74f7.zip | |
refactored naming policy into uri matcher
Diffstat (limited to 'bsie/extractor/image/face/identify.py')
| -rw-r--r-- | bsie/extractor/image/face/identify.py | 11 |
1 files changed, 6 insertions, 5 deletions
diff --git a/bsie/extractor/image/face/identify.py b/bsie/extractor/image/face/identify.py index 152f113..44a75c4 100644 --- a/bsie/extractor/image/face/identify.py +++ b/bsie/extractor/image/face/identify.py @@ -9,7 +9,8 @@ import numpy as np import torch # bsie imports -from bsie.utils import bsfs, node, ns +from bsie.matcher import nodes +from bsie.utils import bsfs, ns # inner-module imports from ... import base @@ -142,10 +143,10 @@ class FaceIdentify(base.Extractor): def extract( self, - subject: node.Node, + subject: nodes.Entity, content: typing.Any, principals: typing.Iterable[bsfs.schema.Predicate], - ) -> typing.Iterator[typing.Tuple[node.Node, bsfs.schema.Predicate, typing.Any]]: + ) -> typing.Iterator[typing.Tuple[nodes.Node, bsfs.schema.Predicate, typing.Any]]: # check principals #if self.schema.predicate(bsf.depicts) not in principals: if self.schema.predicate(ns.bse.face) not in principals: @@ -164,8 +165,8 @@ class FaceIdentify(base.Extractor): lbl = bsfs.URI(self._id2name[idx]) # label (uri) of nearest neighbour if lbl == self._restklasse: # suppress continue - pnode = node.Node(ns.bsn.Person, uri=lbl) - fnode = node.Node(ns.bsn.Face, ucid=face['ucid']) + pnode = nodes.Person(uri=lbl) + fnode = nodes.Face(ucid=face['ucid']) # emit triple yield fnode, self.schema.predicate(bsf.depicts), pnode # FIXME: emit subject -> face -> fnode? |