picpocket / test /test_score.py
chawin.chen
fix gamma值调整评分
8d16a3c
Raw History Blame Contribute Delete
562 Bytes
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("~/Pictures/face/face3.png")
logger.info(
"search results: " + json.dumps(resp, ensure_ascii=False, default=default_converter)
)