diff --git a/run.py b/run.py index 7390c9e..d2075dd 100755 --- a/run.py +++ b/run.py @@ -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)