github/DeepFaceLive
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLive synced 2025-07-30 11:40:37 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactoring

Browse source
This commit is contained in:
iperov 2021-11-07 10:03:15 +04:00
parent 8489949f2c
commit 30ba51edf7
24 changed files with 663 additions and 459 deletions
Download patch file Download diff file Expand all files Collapse all files

4
modelhub/torch/S3FD/S3FD.py
View file

@ -8,13 +8,13 @@ import torch.nn.functional as F
from xlib import math as lib_math
from xlib.file import SplittedFile
from xlib.image import ImageProcessor
from xlib.torch import TorchDeviceInfo, get_cpu_device
from xlib.torch import TorchDeviceInfo, get_cpu_device_info
class S3FD:
def __init__(self, device_info : TorchDeviceInfo = None ):
if device_info is None:
device_info = get_cpu_device()
device_info = get_cpu_device_info()
self.device_info = device_info
path = Path(__file__).parent / 'S3FD.pth'