ZF/insightface
1
0
Fork 0
mirror of https://github.com/deepinsight/insightface.git synced 2026-05-18 22:57:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

tiny on api

Browse Source
This commit is contained in:
Jia Guo
2018-01-24 16:55:49 +08:00
parent c03f510f0b
commit 02f362eb2d
2 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/api/app.py
View File

@@ -84,7 +84,7 @@ def ver():
#print(target_image.shape)
if not isinstance(target_image, list):
target_image = [target_image]
print('before call')
#print('before call')
#ret = model.is_same_id(source_image, target_image)
ret = model.sim(source_image, target_image)
except Exception as ex:
@@ -92,6 +92,7 @@ def ver():
return '-1'
#return str(int(ret))
print('sim', ret)
return "%1.3f"%ret
if __name__ == '__main__':

5
src/api/face_model.py
View File

@@ -65,7 +65,7 @@ class FaceModel:
self.model.sym = all_layers['fc1_output']
def get_aligned_face(self, img, force = False):
print('before det', img.shape)
#print('before det', img.shape)
bounding_boxes, points = detect_face.detect_face(img, self.det_minsize, self.pnet, self.rnet, self.onet, self.det_threshold, self.det_factor)
#if bounding_boxes.shape[0]==0:
# fimg = np.copy(img)
@@ -75,7 +75,7 @@ class FaceModel:
print('force det', img.shape)
bounding_boxes, points = detect_face.detect_face(img, self.det_minsize, self.pnet, self.rnet, self.onet, [0.3, 0.3, 0.1], self.det_factor)
#bounding_boxes, points = detect_face.detect_face_force(img, None, self.pnet, self.rnet, self.onet)
print('after det')
#print('after det')
if bounding_boxes.shape[0]==0:
return None
bindex = 0
@@ -179,6 +179,7 @@ class FaceModel:
return False
def sim(self, source_img, target_img_list):
print('sim start')
source_face = self.get_aligned_face(source_img, True)
print('source face', source_face.shape)
target_face_list = []