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

Fix frames vs. cpu cores issue

Browse Source
This commit is contained in:
henryruhs 2023-06-02 16:38:14 +02:00
parent 8b50aaf838
commit dd5a321c50
1 changed files with 14 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
roop/core.py
View File

@ -96,11 +96,14 @@ def pre_check():
def start_processing():
if args['gpu']:
frame_paths = args["frame_paths"]
n = len(frame_paths) // (args['cores_count'])
# single thread
if args['gpu'] or n < 2:
process_video(args['source_img'], args["frame_paths"])
return
frame_paths = args["frame_paths"]
n = len(frame_paths)//(args['cores_count'])
# multi thread of frames to cpu cores ratio is 2
if n > 2:
processes = []
for i in range(0, len(frame_paths), n):
p = pool.apply_async(process_video, args=(args['source_img'], frame_paths[i:i+n],))