mirror of
https://github.com/serengil/deepface.git
synced 2025-06-07 12:05:22 +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' "
|
"Please install using 'pip install dlib' "
|
||||||
) from e
|
) from e
|
||||||
|
|
||||||
self.layers = [DlibMetaData()]
|
|
||||||
|
|
||||||
# ---------------------
|
|
||||||
|
|
||||||
home = folder_utils.get_deepface_home()
|
home = folder_utils.get_deepface_home()
|
||||||
weight_file = home + "/.deepface/weights/dlib_face_recognition_resnet_model_v1.dat"
|
weight_file = home + "/.deepface/weights/dlib_face_recognition_resnet_model_v1.dat"
|
||||||
|
|
||||||
# ---------------------
|
|
||||||
|
|
||||||
# download pre-trained model if it does not exist
|
# download pre-trained model if it does not exist
|
||||||
if os.path.isfile(weight_file) != True:
|
if os.path.isfile(weight_file) != True:
|
||||||
logger.info("dlib_face_recognition_resnet_model_v1.dat is going to be downloaded")
|
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:
|
with open(newfilepath, "wb") as f:
|
||||||
f.write(data)
|
f.write(data)
|
||||||
|
|
||||||
# ---------------------
|
|
||||||
|
|
||||||
self.model = dlib.face_recognition_model_v1(weight_file)
|
self.model = dlib.face_recognition_model_v1(weight_file)
|
||||||
|
|
||||||
# ---------------------
|
|
||||||
|
|
||||||
# return None # classes must return None
|
# 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."
|
+ "This is an optional dependency."
|
||||||
+ "You can install it as pip install opencv-contrib-python."
|
+ "You can install it as pip install opencv-contrib-python."
|
||||||
) from err
|
) 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