mirror of
https://github.com/serengil/deepface.git
synced 2025-06-07 12:05:22 +00:00
models refactored
This commit is contained in:
parent
d2b9b75a30
commit
b7c7a5580f
@ -1,7 +1,7 @@
|
|||||||
import os
|
import os
|
||||||
import gdown
|
import gdown
|
||||||
import numpy as np
|
import numpy as np
|
||||||
from deepface.basemodels import VGGFace
|
from deepface.models.facial_recognition import VGGFace
|
||||||
from deepface.commons import package_utils, folder_utils
|
from deepface.commons import package_utils, folder_utils
|
||||||
from deepface.models.Demography import Demography
|
from deepface.models.Demography import Demography
|
||||||
from deepface.commons.logger import Logger
|
from deepface.commons.logger import Logger
|
@ -6,7 +6,7 @@ import gdown
|
|||||||
import numpy as np
|
import numpy as np
|
||||||
|
|
||||||
# project dependencies
|
# project dependencies
|
||||||
from deepface.basemodels import VGGFace
|
from deepface.models.facial_recognition import VGGFace
|
||||||
from deepface.commons import package_utils, folder_utils
|
from deepface.commons import package_utils, folder_utils
|
||||||
from deepface.models.Demography import Demography
|
from deepface.models.Demography import Demography
|
||||||
from deepface.commons.logger import Logger
|
from deepface.commons.logger import Logger
|
@ -6,7 +6,7 @@ import gdown
|
|||||||
import numpy as np
|
import numpy as np
|
||||||
|
|
||||||
# project dependencies
|
# project dependencies
|
||||||
from deepface.basemodels import VGGFace
|
from deepface.models.facial_recognition import VGGFace
|
||||||
from deepface.commons import package_utils, folder_utils
|
from deepface.commons import package_utils, folder_utils
|
||||||
from deepface.models.Demography import Demography
|
from deepface.models.Demography import Demography
|
||||||
from deepface.commons.logger import Logger
|
from deepface.commons.logger import Logger
|
@ -4,7 +4,7 @@ import gdown
|
|||||||
import cv2
|
import cv2
|
||||||
import pandas as pd
|
import pandas as pd
|
||||||
import numpy as np
|
import numpy as np
|
||||||
from deepface.detectors import OpenCv
|
from deepface.models.face_detection import OpenCv
|
||||||
from deepface.commons import folder_utils
|
from deepface.commons import folder_utils
|
||||||
from deepface.models.Detector import Detector, FacialAreaRegion
|
from deepface.models.Detector import Detector, FacialAreaRegion
|
||||||
from deepface.commons.logger import Logger
|
from deepface.commons.logger import Logger
|
@ -47,7 +47,7 @@ class Fasnet:
|
|||||||
)
|
)
|
||||||
|
|
||||||
# guarantees Fasnet imported and torch installed
|
# guarantees Fasnet imported and torch installed
|
||||||
from deepface.spoofmodels import FasNetBackbone
|
from deepface.models.spoofing import FasNetBackbone
|
||||||
|
|
||||||
# Fasnet will use 2 distinct models to predict, then it will find the sum of predictions
|
# Fasnet will use 2 distinct models to predict, then it will find the sum of predictions
|
||||||
# to make a final prediction
|
# to make a final prediction
|
@ -7,7 +7,7 @@ from tqdm import tqdm
|
|||||||
|
|
||||||
# project dependencies
|
# project dependencies
|
||||||
from deepface.modules import modeling, detection, preprocessing
|
from deepface.modules import modeling, detection, preprocessing
|
||||||
from deepface.extendedmodels import Gender, Race, Emotion
|
from deepface.models.demography import Gender, Race, Emotion
|
||||||
|
|
||||||
|
|
||||||
def analyze(
|
def analyze(
|
||||||
|
@ -2,7 +2,7 @@
|
|||||||
from typing import Any
|
from typing import Any
|
||||||
|
|
||||||
# project dependencies
|
# project dependencies
|
||||||
from deepface.basemodels import (
|
from deepface.models.facial_recognition import (
|
||||||
VGGFace,
|
VGGFace,
|
||||||
OpenFace,
|
OpenFace,
|
||||||
FbDeepFace,
|
FbDeepFace,
|
||||||
@ -13,7 +13,7 @@ from deepface.basemodels import (
|
|||||||
Facenet,
|
Facenet,
|
||||||
GhostFaceNet,
|
GhostFaceNet,
|
||||||
)
|
)
|
||||||
from deepface.detectors import (
|
from deepface.models.face_detection import (
|
||||||
FastMtCnn,
|
FastMtCnn,
|
||||||
MediaPipe,
|
MediaPipe,
|
||||||
MtCnn,
|
MtCnn,
|
||||||
@ -25,8 +25,8 @@ from deepface.detectors import (
|
|||||||
YuNet,
|
YuNet,
|
||||||
CenterFace,
|
CenterFace,
|
||||||
)
|
)
|
||||||
from deepface.extendedmodels import Age, Gender, Race, Emotion
|
from deepface.models.demography import Age, Gender, Race, Emotion
|
||||||
from deepface.spoofmodels import FasNet
|
from deepface.models.spoofing import FasNet
|
||||||
|
|
||||||
|
|
||||||
def build_model(task: str, model_name: str) -> Any:
|
def build_model(task: str, model_name: str) -> Any:
|
||||||
|
Loading…
x
Reference in New Issue
Block a user