retinaface / handler.py
Edvards's picture
added custom handler
46bc602
raw
history blame contribute delete
321 Bytes
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]