picpocket / test /test_score.py
chenchaoyun
fix
8a25edd
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)
)