mirror of
https://github.com/serengil/deepface.git
synced 2025-06-06 19:45:21 +00:00
layer for external models are no longer being used
This commit is contained in:
parent
dea697c6ec
commit
7753476978
@ -64,15 +64,9 @@ class DlibResNet:
|
||||
"Please install using 'pip install dlib' "
|
||||
) from e
|
||||
|
||||
self.layers = [DlibMetaData()]
|
||||
|
||||
# ---------------------
|
||||
|
||||
home = folder_utils.get_deepface_home()
|
||||
weight_file = home + "/.deepface/weights/dlib_face_recognition_resnet_model_v1.dat"
|
||||
|
||||
# ---------------------
|
||||
|
||||
# download pre-trained model if it does not exist
|
||||
if os.path.isfile(weight_file) != True:
|
||||
logger.info("dlib_face_recognition_resnet_model_v1.dat is going to be downloaded")
|
||||
@ -88,15 +82,6 @@ class DlibResNet:
|
||||
with open(newfilepath, "wb") as f:
|
||||
f.write(data)
|
||||
|
||||
# ---------------------
|
||||
|
||||
self.model = dlib.face_recognition_model_v1(weight_file)
|
||||
|
||||
# ---------------------
|
||||
|
||||
# return None # classes must return None
|
||||
|
||||
|
||||
class DlibMetaData:
|
||||
def __init__(self):
|
||||
self.input_shape = [[1, 150, 150, 3]]
|
||||
|
@ -80,10 +80,3 @@ class SFaceWrapper:
|
||||
+ "This is an optional dependency."
|
||||
+ "You can install it as pip install opencv-contrib-python."
|
||||
) from err
|
||||
|
||||
self.layers = [_Layer()]
|
||||
|
||||
|
||||
class _Layer:
|
||||
input_shape = (None, 112, 112, 3)
|
||||
output_shape = (None, 1, 128)
|
||||
|
Loading…
x
Reference in New Issue
Block a user