Turing311 commited on
Commit
b49fb42
1 Parent(s): de00ac2
Files changed (1) hide show
  1. app.py +7 -3
app.py CHANGED
@@ -192,10 +192,12 @@ def coompare_face_base64():
192
  return response
193
 
194
  feature32_1 = feature1.view(dtype=np.float32)
195
- print(feature32_1)
196
- print('===========================')
197
  feature32 = np.frombuffer(feature1, dtype=np.float32, count=512)
198
- print(feature32)
 
 
199
  similarity = ttv_compare_feature(feature1, feature2)
200
  if similarity > 0.7:
201
  result = "same"
@@ -211,6 +213,8 @@ def coompare_face_base64():
211
  "similarity": float(similarity),
212
  "face1": {"x1": int(faceRect1[0]), "y1": int(faceRect1[1]), "x2": int(faceRect1[2]), "y2" : int(faceRect1[3])},
213
  "face2": {"x1": int(faceRect2[0]), "y1": int(faceRect2[1]), "x2": int(faceRect2[2]), "y2" : int(faceRect2[3])},
 
 
214
  }
215
  })
216
  response.status_code = 200
 
192
  return response
193
 
194
  feature32_1 = feature1.view(dtype=np.float32)
195
+ feature32_2 = feature2.view(dtype=np.float32)
196
+ sim_1 = np.dot(feature32_1, feature32_2)
197
  feature32 = np.frombuffer(feature1, dtype=np.float32, count=512)
198
+ feature322 = np.frombuffer(feature1, dtype=np.float32, count=512)
199
+ sim_2 = np.dot(feature32, feature322)
200
+
201
  similarity = ttv_compare_feature(feature1, feature2)
202
  if similarity > 0.7:
203
  result = "same"
 
213
  "similarity": float(similarity),
214
  "face1": {"x1": int(faceRect1[0]), "y1": int(faceRect1[1]), "x2": int(faceRect1[2]), "y2" : int(faceRect1[3])},
215
  "face2": {"x1": int(faceRect2[0]), "y1": int(faceRect2[1]), "x2": int(faceRect2[2]), "y2" : int(faceRect2[3])},
216
+ "sim_1": float(sim_1),
217
+ "sim_2": float(sim_2),
218
  }
219
  })
220
  response.status_code = 200