Update app.py
Browse files
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(
|
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
|
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(
|
|
|
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)
|