From b9d20c0bff431dd44ce503c0ee45bd9a27e17f60 Mon Sep 17 00:00:00 2001
From: henryruhs <info@henryruhs.com>
Date: Tue, 30 May 2023 01:28:16 +0200
Subject: [PATCH] Move process time to the correct place

---
 core/processor.py |  6 ------
 run.py            | 14 ++++++++++++++
 2 files changed, 14 insertions(+), 6 deletions(-)

diff --git a/core/processor.py b/core/processor.py
index 631feab..d67b1cf 100644
--- a/core/processor.py
+++ b/core/processor.py
@@ -1,6 +1,4 @@
 import os
-import time
-
 import cv2
 import insightface
 import core.globals
@@ -14,7 +12,6 @@ else:
 
 
 def process_video(source_img, frame_paths):
-    start_time = time.time()
     source_face = get_face(cv2.imread(source_img))
     for frame_path in frame_paths:
         frame = cv2.imread(frame_path)
@@ -29,9 +26,6 @@ def process_video(source_img, frame_paths):
         except Exception as e:
             print('E', end='', flush=True)
             pass
-    print(flush=True)
-    end_time = time.time()
-    print(f"Processing time: {end_time - start_time:.2f} seconds", flush=True)
 
 
 def process_img(source_img, target_path):
diff --git a/run.py b/run.py
index cd543a2..4245c39 100755
--- a/run.py
+++ b/run.py
@@ -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():