添加测试方法

This commit is contained in:
haotianmingyue 2025-09-19 10:04:06 +08:00
parent 8fcfeafcbd
commit 611098b5e2

View File

@ -51,23 +51,23 @@ async def get_face_group_service():
f.write(json.dumps(result[1]))
# 人脸分组1vN搜索
async def face_group_1vN_search_service(image_path):
async def face_group_1vN_search_service(image_path,faceGroupIndexCodes:list, pageNo=1, pageSize=5, searchNum:int=5 , minSimilarity=50):
with open(image_path, 'rb') as f:
image_data = f.read()
encoded_image = base64.b64encode(image_data).decode('utf-8')
with open("image_base64.txt", "w", encoding="utf-8") as f:
f.write(encoded_image)
# with open("image_base64.txt", "w", encoding="utf-8") as f:
# f.write(encoded_image)
result = await HaikangUtil.face_group_1vN_search(
facePicBinaryData=encoded_image,
pageNo=1,
pageSize=10,
searchNum=99,
minSimilarity=50,
faceGroupIndexCodes=['70cd3fcdcdb444a0b92e75ff66b549e6']
pageNo=pageNo,
pageSize=pageSize,
searchNum=searchNum,
minSimilarity=minSimilarity,
faceGroupIndexCodes=faceGroupIndexCodes
)
print(result)
with open("face_group_1vN_search_service.json", "w", encoding="utf-8") as f:
@ -240,3 +240,6 @@ if __name__ == '__main__':
# 向分组中添加单个人脸
asyncio.run(face_single_addition_service("d4a610b6554d4544bc35dd3138f128b0", "./a7b3c794281902041c9b823df7667ce7.jpg", "测试"))
# 人脸分组1vN搜索
asyncio.run(face_group_1vN_search_service("./测试图片1.jpg", ["d4a610b6554d4544bc35dd3138f128b0"]))