apps/roop
1
0
Fork 0
mirror of https://github.com/s0md3v/roop.git synced 2025-10-05 08:17:00 +08:00
Code Issues Actions 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
committed by Somdev Sangwan
parent d7f72f7c5a
commit b9d20c0bff
2 changed files with 14 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
run.py
View File

@@ -3,12 +3,15 @@
<<<<<<< HEAD
import sys
import time
<<<<<<< HEAD
=======
>>>>>>> 54f800d (Add GPU support, Quit on missing model, Remove globals (sorry))
=======
import sys
>>>>>>> dc0653a (Restore globals, add process time for better comparison)
=======
>>>>>>> 2b14613 (Move process time to the correct place)
import torch
import shutil
import core.globals
@@ -61,8 +64,13 @@ def start_processing():
start_time = time.time()
if args['gpu']:
process_video(args['source_img'], args["frame_paths"])
<<<<<<< HEAD
end_time = time.time()
print(flush=True)
=======
print(flush=True)
end_time = time.time()
>>>>>>> 2b14613 (Move process time to the correct place)
print(f"Processing time: {end_time - start_time:.2f} seconds", flush=True)
return
frame_paths = args["frame_paths"]
@@ -75,6 +83,7 @@ def start_processing():
p.get()
pool.close()
pool.join()
<<<<<<< HEAD
end_time = time.time()
print(flush=True)
print(f"Processing time: {end_time - start_time:.2f} seconds", flush=True)
@@ -108,6 +117,11 @@ def preview_video(video_path):
img_label.pack()
cap.release()
=======
print(flush=True)
end_time = time.time()
print(f"Processing time: {end_time - start_time:.2f} seconds", flush=True)
>>>>>>> 2b14613 (Move process time to the correct place)
def select_face():