github/DeepFaceLive
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLive synced 2025-07-06 04:52:14 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

_

Browse source
This commit is contained in:
iperov 2021-11-16 15:44:14 +04:00
parent 003afa02cf
commit 765a72140f
1 changed files with 5 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

8
apps/trainers/FaceAligner/FaceAlignerTrainerApp.py
View file

@ -7,16 +7,18 @@ import cv2
import numpy as np import numpy as np
import torch import torch
import torch.autograd import torch.autograd
import torchvision.models as tv
from localization import L, Localization from localization import L, Localization
from modelhub import torch as torch_models from modelhub import torch as torch_models
from xlib import math as lib_math
from xlib import torch as lib_torch from xlib import torch as lib_torch
from xlib.console import diacon as dc from xlib.console import diacon as dc
from xlib.torch.optim import AdaBelief
from xlib.torch.device import TorchDeviceInfo from xlib.torch.device import TorchDeviceInfo
import torchvision.models as tv from xlib.torch.optim import AdaBelief
from xlib import math as lib_math
from .TrainingDataGenerator import Data, TrainingDataGenerator from .TrainingDataGenerator import Data, TrainingDataGenerator
class FaceAlignerTrainerApp: class FaceAlignerTrainerApp:
def __init__(self, workspace_path : Path, faceset_path : Path): def __init__(self, workspace_path : Path, faceset_path : Path):
print('Initializing trainer.\n') print('Initializing trainer.\n')