justin2341 commited on
Commit
8da3e7b
·
verified ·
1 Parent(s): b2c192d

Update demo.py

Browse files
Files changed (1) hide show
  1. demo.py +11 -10
demo.py CHANGED
@@ -29,24 +29,25 @@ def face_crop(image, face_rect):
29
  face_image = face_image.resize((int(resized_w), int(resized_h)))
30
  return face_image
31
 
32
- def compare_face(image1, image2):
 
 
 
 
 
33
  try:
34
- img_bytes1 = io.BytesIO()
35
- image1.save(img_bytes1, format="JPEG")
36
- img_bytes1.seek(0)
37
  except:
38
  return "Failed to open image1"
39
 
40
  try:
41
- img_bytes2 = io.BytesIO()
42
- image2.save(img_bytes2, format="JPEG")
43
- img_bytes2.seek(0)
44
  except:
45
  return "Failed to open image2"
46
 
47
- # url = "http://127.0.0.1:8080/compare_face"
48
- # files = {'file1': img_bytes1, 'file2': img_bytes2}
49
- # result = requests.post(url=url, files=files)
50
 
51
  return "abcdef"
52
 
 
29
  face_image = face_image.resize((int(resized_w), int(resized_h)))
30
  return face_image
31
 
32
+ def compare_face(frame1, frame2):
33
+ url = "http://127.0.0.1:8080/compare_face"
34
+ files = {'file1': open(frame1, 'rb'), 'file2': open(frame2, 'rb')}
35
+
36
+ file1 = None
37
+ file2 = None
38
  try:
39
+ file1 = open(frame1, 'rb')
 
 
40
  except:
41
  return "Failed to open image1"
42
 
43
  try:
44
+ file2 = open(frame2, 'rb')
 
 
45
  except:
46
  return "Failed to open image2"
47
 
48
+ url = "http://127.0.0.1:8080/compare_face"
49
+ files = {'file1': file1, 'file2': file2}
50
+ result = requests.post(url=url, files=files)
51
 
52
  return "abcdef"
53