Примеры

В следующих примерах наглядно продемонстрированы описанные выше принципы написания плагина, а также использование классов и методов.

  1. В данном примере плагин запрашивает извлечение вектора признаков и информации об эмоциях, если обнаруженное лицо содержит метку ‘emo’, и записывает все полученную информацию в лог.

    import logging
    
    from ntech import sfapi_client
    
    from facerouter.plugin import Plugin
    
    logger = logging.getLogger(__name__)
    
    
    class LogEmoPlugin(Plugin):
        async def preprocess(self, request, labels):
            if labels.get('emo'):
                return ('facen', 'emotions')
    
        async def process(self, request, photo, bbox, event_id, detection: sfapi_client.DetectFace):
            logger.info('%r: %r', bbox, detection.features.get('emotions')[0]['emotion'])
            logger.info('%r: params: ', bbox)
            for param in request.params._fields:
                param_repr = repr(getattr(request.params, param))
                if len(param_repr) > 100:
                    param_repr = param_repr[:97] + "..."
                logger.info("%r: %s", param, param_repr)
    
    
    def activate(app, ctx, plugin_name, plugin_source):
        return LogEmoPlugin(ctx=ctx)
    
  2. В данном примере плагин запрашивает извлечение вектора признаков, после этого сохраняет лицо в галерею ’ppl’ Tarantool. Если такой галереи не существует, она будет создана.

    import logging
    import PIL.Image
    import time
    from io import BytesIO
    
    from ntech import sfapi_client
    
    from facerouter.plugin import Plugin
    
    logger = logging.getLogger(__name__)
    
    
    class EnrollPlugin(Plugin):
        async def preprocess(self, request, labels):
            if labels.get('lol') == 'kek':
                return ('facen',)
    
        async def process(self, request, photo, bbox, event_id, detection: sfapi_client.DetectFace):
            img = PIL.Image.open(BytesIO(photo))
            thumb = img.crop(bbox)
            fname = '/tmp/%x.jpeg' % (event_id,)
            thumb.save(fname)
            while True:
                try:
                    await self.ctx.sfapi['ppl'].add(event_id, detection, meta={
                        'timestamp': int(time.time()),
                        'photo_hash': fname,
                    })
                except sfapi_client.SFApiRemoteError as e:
                    if e.code == "GALLERY_NOT_FOUND":
                        await self.ctx.sfapi['ppl'].create()
                    else:
                        raise
                else:
                    break
            logger.info('%r: %r %r', bbox, event_id, fname)
    
    
    def activate(app, ctx, plugin_name, plugin_source):
        return EnrollPlugin(ctx=ctx)