upagge/roop
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Move process time to the correct place

Browse Source
This commit is contained in:
henryruhs 2023-05-30 01:28:16 +02:00
parent dc0653ad39
commit 2b14613a46
2 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
core/processor.py
View File

@ -1,6 +1,4 @@
import os import os
import time
import cv2 import cv2
import insightface import insightface
import core.globals import core.globals
@ -14,7 +12,6 @@ else:
def process_video(source_img, frame_paths): def process_video(source_img, frame_paths):
start_time = time.time()
source_face = get_face(cv2.imread(source_img)) source_face = get_face(cv2.imread(source_img))
for frame_path in frame_paths: for frame_path in frame_paths:
frame = cv2.imread(frame_path) frame = cv2.imread(frame_path)
@ -29,9 +26,6 @@ def process_video(source_img, frame_paths):
except Exception as e: except Exception as e:
print('E', end='', flush=True) print('E', end='', flush=True)
pass pass
print(flush=True)
end_time = time.time()
print(f"Processing time: {end_time - start_time:.2f} seconds", flush=True)
def process_img(source_img, target_path): def process_img(source_img, target_path):

8
run.py
View File

@ -1,5 +1,6 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
import sys import sys
import time
import torch import torch
import shutil import shutil
import core.globals import core.globals
@ -47,8 +48,12 @@ if os.name == "nt":
def start_processing(): def start_processing():
start_time = time.time()
if args['gpu']: if args['gpu']:
process_video(args['source_img'], args["frame_paths"]) process_video(args['source_img'], args["frame_paths"])
print(flush=True)
end_time = time.time()
print(f"Processing time: {end_time - start_time:.2f} seconds", flush=True)
return return
frame_paths = args["frame_paths"] frame_paths = args["frame_paths"]
n = len(frame_paths)//(psutil.cpu_count()-1) n = len(frame_paths)//(psutil.cpu_count()-1)
@ -60,6 +65,9 @@ def start_processing():
p.get() p.get()
pool.close() pool.close()
pool.join() pool.join()
print(flush=True)
end_time = time.time()
print(f"Processing time: {end_time - start_time:.2f} seconds", flush=True)
def select_face(): def select_face():