mirror of
https://github.com/iperov/DeepFaceLive
synced 2025-07-05 20:42:12 -07:00
93 lines
2.5 KiB
Python
93 lines
2.5 KiB
Python
import os
|
|
import platform
|
|
|
|
# onnxruntime==1.8.0 requires CUDA_PATH_V11_2, but 1.8.1 don't
|
|
# keep the code if they return that behaviour
|
|
# if __name__ == '__main__':
|
|
# if platform.system() == 'Windows':
|
|
# if 'CUDA_PATH' not in os.environ:
|
|
# raise Exception('CUDA_PATH should be set to environ')
|
|
# # set environ for onnxruntime
|
|
# # os.environ['CUDA_PATH_V11_2'] = os.environ['CUDA_PATH']
|
|
|
|
import argparse
|
|
from pathlib import Path
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser()
|
|
subparsers = parser.add_subparsers()
|
|
|
|
run_parser = subparsers.add_parser( "run", help="Run the application.")
|
|
|
|
run_subparsers = run_parser.add_subparsers()
|
|
|
|
def run_DeepFaceLive(args):
|
|
userdata_path = Path(args.userdata_dir)
|
|
print('Running DeepFaceLive.')
|
|
from apps.DeepFaceLive.DeepFaceLiveApp import DeepFaceLiveApp
|
|
DeepFaceLiveApp(userdata_path=userdata_path).run()
|
|
|
|
p = run_subparsers.add_parser('DeepFaceLive')
|
|
p.add_argument('--userdata-dir', default=None, action=fixPathAction, help="Workspace directory.")
|
|
p.set_defaults(func=run_DeepFaceLive)
|
|
|
|
def bad_args(arguments):
|
|
parser.print_help()
|
|
exit(0)
|
|
parser.set_defaults(func=bad_args)
|
|
|
|
args = parser.parse_args()
|
|
args.func(args)
|
|
|
|
class fixPathAction(argparse.Action):
|
|
def __call__(self, parser, namespace, values, option_string=None):
|
|
setattr(namespace, self.dest, os.path.abspath(os.path.expanduser(values)))
|
|
|
|
|
|
# from xlib import time as lib_time
|
|
# import numpy as np
|
|
# import cv2
|
|
# import cupy as cp
|
|
# import cupyx.scipy.ndimage
|
|
# import scipy
|
|
# import scipy.ndimage
|
|
|
|
# from typing import Union, Tuple
|
|
|
|
# from xlib.image import ImageProcessor
|
|
|
|
# mat = def_mat = np.array([[ 8.5966533e-01, 8.3356246e-02, 1.9525000e+02 ],#
|
|
# [-8.3356142e-02, 8.5966533e-01, 8.8052826e+01 ]], np.float32)#
|
|
|
|
# is_cp = False
|
|
# while True:
|
|
# print('is_cp : ', is_cp)
|
|
# img = cv2.imread(r'D:\DevelopPython\test\00000.png')
|
|
# if is_cp:
|
|
# img = cp.asarray(img)
|
|
# is_cp = not is_cp
|
|
|
|
# ip = ImageProcessor(img)
|
|
# ip.sharpen(factor=10.0)
|
|
# #ip.degrade_resize( np.random.rand() )
|
|
|
|
|
|
# #ip.erode_blur(50, 50, fade_to_border=False)
|
|
# #ip.resize( (500,500) )
|
|
# #ip.warpAffine(mat, 1920, 1080)
|
|
# x = ip.get_image('HWC')
|
|
|
|
# x = cp.asnumpy(x)
|
|
# cv2.imshow('', x )
|
|
# cv2.waitKey(0)
|
|
# import code
|
|
# code.interact(local=dict(globals(), **locals()))
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|
|
|
|
|
|
# import code
|
|
# code.interact(local=dict(globals(), **locals()))
|