Move process time to the correct place

This commit is contained in:
henryruhs 2023-05-30 01:28:42 +02:00
parent 2b14613a46
commit 0f782ff516

4
run.py
View File

@ -51,8 +51,8 @@ def start_processing():
start_time = time.time()
if args['gpu']:
process_video(args['source_img'], args["frame_paths"])
print(flush=True)
end_time = time.time()
print(flush=True)
print(f"Processing time: {end_time - start_time:.2f} seconds", flush=True)
return
frame_paths = args["frame_paths"]
@ -65,8 +65,8 @@ def start_processing():
p.get()
pool.close()
pool.join()
print(flush=True)
end_time = time.time()
print(flush=True)
print(f"Processing time: {end_time - start_time:.2f} seconds", flush=True)