fi0r1/roop
1
0
Fork 0
mirror of https://github.com/s0md3v/roop.git synced 2025-12-07 10:28:28 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix namespace from core to roop

Browse Source
This commit is contained in:
Jose Manuel 2023-06-02 20:29:37 +02:00
parent 6e1045ee92
commit b7ce758c99
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
roop/core.py
View File

@ -75,7 +75,7 @@ def pre_check():
if not os.path.isfile(model_path): if not os.path.isfile(model_path):
quit('File "inswapper_128.onnx" does not exist!') quit('File "inswapper_128.onnx" does not exist!')
if '--gpu' in sys.argv: if '--gpu' in sys.argv:
core.globals.use_gpu = True roop.globals.use_gpu = True
NVIDIA_PROVIDERS = ['CUDAExecutionProvider', 'TensorrtExecutionProvider'] NVIDIA_PROVIDERS = ['CUDAExecutionProvider', 'TensorrtExecutionProvider']
if len(list(set(roop.globals.providers) - set(NVIDIA_PROVIDERS))) == 1: if len(list(set(roop.globals.providers) - set(NVIDIA_PROVIDERS))) == 1:
CUDA_VERSION = torch.version.cuda CUDA_VERSION = torch.version.cuda
@ -100,7 +100,7 @@ def start_processing():
frame_paths = args["frame_paths"] frame_paths = args["frame_paths"]
n = len(frame_paths) // (args['cores_count']) n = len(frame_paths) // (args['cores_count'])
# single thread # single thread
if core.globals.use_gpu or n < 2: if roop.globals.use_gpu or n < 2:
process_video(args['source_img'], args["frame_paths"]) process_video(args['source_img'], args["frame_paths"])
return return
# multithread if total frames to cpu cores ratio is greater than 2 # multithread if total frames to cpu cores ratio is greater than 2
@ -217,12 +217,12 @@ def start():
fps, exact_fps = detect_fps(target_path) fps, exact_fps = detect_fps(target_path)
if not args['keep_fps'] and fps > 30: if not args['keep_fps'] and fps > 30:
this_path = output_dir + "/" + video_name + ".mp4" this_path = output_dir + "/" + video_name + ".mp4"
set_fps(target_path, this_path, 30, core.globals.use_gpu) set_fps(target_path, this_path, 30, roop.globals.use_gpu)
target_path, exact_fps = this_path, 30 target_path, exact_fps = this_path, 30
else: else:
shutil.copy(target_path, output_dir) shutil.copy(target_path, output_dir)
status("extracting frames...") status("extracting frames...")
extract_frames(target_path, output_dir, core.globals.use_gpu) extract_frames(target_path, output_dir, roop.globals.use_gpu)
args['frame_paths'] = tuple(sorted( args['frame_paths'] = tuple(sorted(
glob.glob(output_dir + "/*.png"), glob.glob(output_dir + "/*.png"),
key=lambda x: int(x.split(sep)[-1].replace(".png", "")) key=lambda x: int(x.split(sep)[-1].replace(".png", ""))
@ -230,9 +230,9 @@ def start():
status("swapping in progress...") status("swapping in progress...")
start_processing() start_processing()
status("creating video...") status("creating video...")
create_video(video_name, exact_fps, output_dir, core.globals.use_gpu) create_video(video_name, exact_fps, output_dir, roop.globals.use_gpu)
status("adding audio...") status("adding audio...")
add_audio(output_dir, target_path, video_name_full, args['keep_frames'], args['output_file'], core.globals.use_gpu) add_audio(output_dir, target_path, video_name_full, args['keep_frames'], args['output_file'], roop.globals.use_gpu)
save_path = args['output_file'] if args['output_file'] else output_dir + "/" + video_name + ".mp4" save_path = args['output_file'] if args['output_file'] else output_dir + "/" + video_name + ".mp4"
print("\n\nVideo saved as:", save_path, "\n\n") print("\n\nVideo saved as:", save_path, "\n\n")
status("swap successful!") status("swap successful!")