1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
# coding: utf-8
# -------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for
# license information.
# --------------------------------------------------------------------------
from devtools_testutils import AzureRecordedTestCase
from devtools_testutils.aio import recorded_by_proxy_async
from azure.ai.vision.face.aio import FaceClient, FaceAdministrationClient
from azure.ai.vision.face.models import (
FaceDetectionModel,
FaceRecognitionModel,
)
from preparers import AsyncFaceClientPreparer, FacePreparer, AsyncFaceAdministrationClientPreparer
from _shared.constants import TestImages
from _shared import helpers
class TestFindSimilarAsync(AzureRecordedTestCase):
test_images = [TestImages.IMAGE_FAMILY_1_MOM_1, TestImages.IMAGE_FAMILY_2_LADY_1]
list_id = "findsimilar"
async def _setup_faces(self, client: FaceClient):
face_ids = []
for image in self.test_images:
image_path = helpers.get_image_path(image)
result = await client.detect(
helpers.read_file_content(image_path),
detection_model=FaceDetectionModel.DETECTION03,
recognition_model=FaceRecognitionModel.RECOGNITION04,
return_face_id=True,
)
face_ids.append(result[0].face_id)
return face_ids
async def _setup_large_face_list(self, client: FaceAdministrationClient):
operations = client.large_face_list
await operations.create(self.list_id, name=self.list_id, recognition_model=FaceRecognitionModel.RECOGNITION04)
persisted_face_ids = []
for image in self.test_images:
image_path = helpers.get_image_path(image)
result = await operations.add_face(
self.list_id, helpers.read_file_content(image_path), detection_model=FaceDetectionModel.DETECTION03
)
assert result.persisted_face_id
persisted_face_ids.append(result.persisted_face_id)
poller = await operations.begin_train(self.list_id, polling_interval=3)
await poller.wait()
return persisted_face_ids
# TODO: Use fixtures to replace teardown methods
async def _teardown_large_face_list(self, client: FaceAdministrationClient):
await client.large_face_list.delete(self.list_id)
@FacePreparer()
@AsyncFaceClientPreparer()
@AsyncFaceAdministrationClientPreparer()
@recorded_by_proxy_async
async def test_find_similar_from_large_face_list(
self, client: FaceClient, administration_client: FaceAdministrationClient
):
face_ids = await self._setup_faces(client)
persisted_face_ids = await self._setup_large_face_list(administration_client)
similar_faces = await client.find_similar_from_large_face_list(
face_id=face_ids[0], large_face_list_id=self.list_id
)
assert similar_faces[0].persisted_face_id == persisted_face_ids[0]
assert similar_faces[0].confidence > 0.9
await self._teardown_large_face_list(administration_client)
|
