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:42 +02:00
parent 2b14613a46
commit 0f782ff516
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
run.py
View File

@ -51,8 +51,8 @@ def start_processing():
start_time = time.time() 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() end_time = time.time()
print(flush=True)
print(f"Processing time: {end_time - start_time:.2f} seconds", flush=True) print(f"Processing time: {end_time - start_time:.2f} seconds", flush=True)
return return
frame_paths = args["frame_paths"] frame_paths = args["frame_paths"]
@ -65,8 +65,8 @@ def start_processing():
p.get() p.get()
pool.close() pool.close()
pool.join() pool.join()
print(flush=True)
end_time = time.time() end_time = time.time()
print(flush=True)
print(f"Processing time: {end_time - start_time:.2f} seconds", flush=True) print(f"Processing time: {end_time - start_time:.2f} seconds", flush=True)