Skip to content

Commit

Permalink
no need to check version for fb deepface
Browse files Browse the repository at this point in the history
  • Loading branch information
serengil committed Mar 19, 2024
1 parent 87a95dc commit ce1fa57
Showing 1 changed file with 2 additions and 15 deletions.
17 changes: 2 additions & 15 deletions deepface/basemodels/FbDeepFace.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,7 @@
Flatten,
Dense,
Dropout,
LocallyConnected2D,
)
else:
from tensorflow.keras.models import Model, Sequential
Expand All @@ -32,6 +33,7 @@
Flatten,
Dense,
Dropout,
LocallyConnected2D,
)


Expand All @@ -43,14 +45,6 @@ class DeepFaceClient(FacialRecognition):
"""

def __init__(self):
# DeepFace requires tf 2.12 or less
if tf_major == 2 and tf_minor > 12:
# Ref: https://github.com/serengil/deepface/pull/1079
raise ValueError(
"DeepFace model requires LocallyConnected2D but it is no longer supported"
f" after tf 2.12 but you have {tf_major}.{tf_minor}. You need to downgrade your tf."
)

self.model = load_model()
self.model_name = "DeepFace"
self.input_shape = (152, 152)
Expand All @@ -75,13 +69,6 @@ def load_model(
"""
Construct DeepFace model, download its weights and load
"""
# we have some checks for this dependency in the init of client
# putting this in global causes library initialization
if tf_major == 1:
from keras.layers import LocallyConnected2D
else:
from tensorflow.keras.layers import LocallyConnected2D

base_model = Sequential()
base_model.add(
Convolution2D(32, (11, 11), activation="relu", name="C1", input_shape=(152, 152, 3))
Expand Down

0 comments on commit ce1fa57

Please sign in to comment.