Примеры

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

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)