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

comments for resize

Browse Source
This commit is contained in:
Sefik Ilkin Serengil 2021-07-01 17:17:46 +03:00 committed by GitHub
parent 6b2f343eea
commit 94236318ac
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
deepface/commons/functions.py
View File

@ -123,11 +123,14 @@ def preprocess_face(img, target_size=(224, 224), grayscale = False, enforce_dete
#post-processing
if grayscale == True:
img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
#---------------------------------------------------
#resize image to expected shape
# img = cv2.resize(img, target_size)
# img = cv2.resize(img, target_size) #resize causes transformation on base image, adding black pixels to resize will not deform the base image
# First resize the longer side to the target size
factor = target_size[0] / max(img.shape) # Suppose target_size[0] == target_size[1]
factor = target_size[0] / max(img.shape) # Suppose target_size[0] == target_size[1]. What if this condition is not satisfied? E.g. DeepID expects 57x47 shaped inputs.
dsize = (int(img.shape[1] * factor), int(img.shape[0] * factor))
img = cv2.resize(img, dsize)
@ -140,6 +143,8 @@ def preprocess_face(img, target_size=(224, 224), grayscale = False, enforce_dete
else:
img = np.pad(img, ((diff_0 // 2, diff_0 - diff_0 // 2), (diff_1 // 2, diff_1 - diff_1 // 2)), 'constant')
#---------------------------------------------------
img_pixels = image.img_to_array(img)
img_pixels = np.expand_dims(img_pixels, axis = 0)
img_pixels /= 255 #normalize input in [0, 1]