github/deepface
1
1
Fork 0
mirror of https://github.com/serengil/deepface.git synced 2025-06-08 20:45:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

resolving issue 1097

Browse Source
This commit is contained in:
Sefik Ilkin Serengil 2024-03-11 18:31:52 +00:00
parent 049ab8cdff
commit 9705e6f4ef
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
deepface/detectors/YuNet.py
View File

@ -74,9 +74,9 @@ class YuNetClient(Detector):
# resize image if it is too large (Yunet fails to detect faces on large input sometimes) # resize image if it is too large (Yunet fails to detect faces on large input sometimes)
# I picked 640 as a threshold because it is the default value of max_size in Yunet. # I picked 640 as a threshold because it is the default value of max_size in Yunet.
resized = False resized = False
r = 1 # resize factor
if height > 640 or width > 640: if height > 640 or width > 640:
r = 640.0 / max(height, width) r = 640.0 / max(height, width)
original_image = img.copy()
img = cv2.resize(img, (int(width * r), int(height * r))) img = cv2.resize(img, (int(width * r), int(height * r)))
height, width = img.shape[0], img.shape[1] height, width = img.shape[0], img.shape[1]
resized = True resized = True
@ -109,7 +109,6 @@ class YuNetClient(Detector):
x = max(x, 0) x = max(x, 0)
y = max(y, 0) y = max(y, 0)
if resized: if resized:
img = original_image
x, y, w, h = int(x / r), int(y / r), int(w / r), int(h / r) x, y, w, h = int(x / r), int(y / r), int(w / r), int(h / r)
x_re, y_re, x_le, y_le = ( x_re, y_re, x_le, y_le = (
int(x_re / r), int(x_re / r),