vialcald commited on
Commit
f870e1d
1 Parent(s): 3e27397

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -23,7 +23,7 @@ def dhash(image, hashSize=8):
23
  return sum([2 ** i for (i, v) in enumerate(diff.flatten()) if v])
24
 
25
  def face_detector(image):
26
- image = PILImage.create(img)
27
  image = imutils.resize(image, width=500)
28
  gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
29
  l=[]
@@ -34,14 +34,13 @@ def face_detector(image):
34
  l.append(image[y:y+h, x:x+w])
35
  return l
36
 
37
-
38
  def searchAll(img):
39
  sol = []
40
  for face in face_detector(img):
41
  sol + search(face)
42
  return sol
43
 
44
- # Definimos una función que se encarga de llevar a cabo las búsquedas
45
  def search(img, dist=10):
46
  queryHash = convert_hash(dhash(img))
47
  treeResults = sorted(tree.get_all_in_range(queryHash, dist))
@@ -49,4 +48,4 @@ def search(img, dist=10):
49
  return [cv2.imread(hashes.get(h, [])) for (d, h) in results]
50
 
51
  # Creamos la interfaz y la lanzamos.
52
- gr.Interface(fn=searchAll, inputs=gr.inputs.Image(shape=(128, 128)), outputs=gr.Gallery()).launch(share=False)
 
23
  return sum([2 ** i for (i, v) in enumerate(diff.flatten()) if v])
24
 
25
  def face_detector(image):
26
+ image = PILImage.create(image)
27
  image = imutils.resize(image, width=500)
28
  gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
29
  l=[]
 
34
  l.append(image[y:y+h, x:x+w])
35
  return l
36
 
 
37
  def searchAll(img):
38
  sol = []
39
  for face in face_detector(img):
40
  sol + search(face)
41
  return sol
42
 
43
+ # Definimos una función que se encarga de llevar a cabo cada búsqueda
44
  def search(img, dist=10):
45
  queryHash = convert_hash(dhash(img))
46
  treeResults = sorted(tree.get_all_in_range(queryHash, dist))
 
48
  return [cv2.imread(hashes.get(h, [])) for (d, h) in results]
49
 
50
  # Creamos la interfaz y la lanzamos.
51
+ gr.Interface(fn=searchAll, inputs=gr.inputs.Image(), outputs=gr.Gallery()).launch(share=False)