From bc7910e79fb592d27abc628bbca20736c2794a7b Mon Sep 17 00:00:00 2001
From: henryruhs <info@henryruhs.com>
Date: Mon, 16 Oct 2023 15:46:29 +0200
Subject: [PATCH] Uniform frame handling for predictor

---
 facefusion/predictor.py | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/facefusion/predictor.py b/facefusion/predictor.py
index 24040135..52962f51 100644
--- a/facefusion/predictor.py
+++ b/facefusion/predictor.py
@@ -61,11 +61,16 @@ def predict_stream(frame : Frame, fps : float) -> bool:
 	return False
 
 
+def prepare_frame(frame : Frame) -> Frame:
+	frame = cv2.resize(frame, (224, 224)).astype(numpy.float32)
+	frame -= numpy.array([ 104, 117, 123 ]).astype(numpy.float32)
+	frame = numpy.expand_dims(frame, axis = 0)
+	return frame
+
+
 def predict_frame(frame : Frame) -> bool:
 	predictor = get_predictor()
-	frame = cv2.resize(frame, (224, 224)).astype(numpy.float32)
-	frame -= numpy.array([ 104, 117, 123 ], dtype = numpy.float32)
-	frame = numpy.expand_dims(frame, axis = 0)
+	frame = prepare_frame(frame)
 	probability = predictor.run(None,
 	{
 		'input:0': frame