retinaface / handler.py
Edvards's picture
added custom handler
46bc602
from typing import Dict, List, Any
from retinaface import RetinaFace
import numpy as np
from PIL import Image
class EndpointHandler():
def __init__(self, path=""):
return
def __call__(self, data: np.ndarray) -> List[Dict[str, Any]]:
res = RetinaFace.detect_faces(data)
return [res]