test/test_score.py

import json
import logging

import numpy as np
from retinaface import RetinaFace


def default_converter(o):
    if isinstance(o, np.integer):
        return int(o)
    if isinstance(o, np.floating):
        return float(o)
    if isinstance(o, np.ndarray):
        return o.tolist()
    return str(o)


# 配置日志
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

resp = RetinaFace.detect_faces("~/Downloads/chounan.jpeg")

logger.info(
    "search results: " + json.dumps(resp, ensure_ascii=False, default=default_converter)
)