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 torch
import torch.autograd
import torchvision.models as tv
from localization import L, Localization
from modelhub import torch as torch_models
from xlib import math as lib_math
from xlib import torch as lib_torch
from xlib.console import diacon as dc
from xlib.torch.optim import AdaBelief
from xlib.torch.device import TorchDeviceInfo
import torchvision.models as tv
from xlib import math as lib_math
from xlib.torch.optim import AdaBelief
from .TrainingDataGenerator import Data, TrainingDataGenerator
class FaceAlignerTrainerApp:
def __init__(self, workspace_path : Path, faceset_path : Path):
print('Initializing trainer.\n')