models refactored

This commit is contained in:
Sefik Ilkin Serengil 2024-08-08 08:53:56 +01:00
parent d2b9b75a30
commit b7c7a5580f
31 changed files with 10 additions and 10 deletions

View File

@ -1,7 +1,7 @@
import os
import gdown
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.models.Demography import Demography
from deepface.commons.logger import Logger

View File

@ -6,7 +6,7 @@ import gdown
import numpy as np
# project dependencies
from deepface.basemodels import VGGFace
from deepface.models.facial_recognition import VGGFace
from deepface.commons import package_utils, folder_utils
from deepface.models.Demography import Demography
from deepface.commons.logger import Logger

View File

@ -6,7 +6,7 @@ import gdown
import numpy as np
# project dependencies
from deepface.basemodels import VGGFace
from deepface.models.facial_recognition import VGGFace
from deepface.commons import package_utils, folder_utils
from deepface.models.Demography import Demography
from deepface.commons.logger import Logger

View File

@ -4,7 +4,7 @@ import gdown
import cv2
import pandas as pd
import numpy as np
from deepface.detectors import OpenCv
from deepface.models.face_detection import OpenCv
from deepface.commons import folder_utils
from deepface.models.Detector import Detector, FacialAreaRegion
from deepface.commons.logger import Logger

View File

@ -47,7 +47,7 @@ class Fasnet:
)
# 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
# to make a final prediction

View File

@ -7,7 +7,7 @@ from tqdm import tqdm
# project dependencies
from deepface.modules import modeling, detection, preprocessing
from deepface.extendedmodels import Gender, Race, Emotion
from deepface.models.demography import Gender, Race, Emotion
def analyze(

View File

@ -2,7 +2,7 @@
from typing import Any
# project dependencies
from deepface.basemodels import (
from deepface.models.facial_recognition import (
VGGFace,
OpenFace,
FbDeepFace,
@ -13,7 +13,7 @@ from deepface.basemodels import (
Facenet,
GhostFaceNet,
)
from deepface.detectors import (
from deepface.models.face_detection import (
FastMtCnn,
MediaPipe,
MtCnn,
@ -25,8 +25,8 @@ from deepface.detectors import (
YuNet,
CenterFace,
)
from deepface.extendedmodels import Age, Gender, Race, Emotion
from deepface.spoofmodels import FasNet
from deepface.models.demography import Age, Gender, Race, Emotion
from deepface.models.spoofing import FasNet
def build_model(task: str, model_name: str) -> Any: