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/detect.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/detect.py')
-rw-r--r-- | bsie/extractor/image/face/detect.py | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/bsie/extractor/image/face/detect.py b/bsie/extractor/image/face/detect.py index 94e3a61..51d5659 100644 --- a/bsie/extractor/image/face/detect.py +++ b/bsie/extractor/image/face/detect.py @@ -7,7 +7,8 @@ import torch from facenet_pytorch import MTCNN, InceptionResnetV1 # 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 @@ -72,17 +73,17 @@ class FaceDetect(base.Extractor): def extract( self, - subject: node.Node, + subject: nodes.Entity, content: dict, 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(ns.bse.face) not in principals: # nothing to do; abort return for face in content: - fnode = node.Node(ns.bsn.Face, ucid=face['ucid']) + fnode = nodes.Face(ucid=face['ucid']) yield subject, ns.bse.face, fnode yield fnode, bsf.x, face['x'] yield fnode, bsf.y, face['y'] |