mirrors/facefusion
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Uniform frame handling for predictor

Browse Source
This commit is contained in:
henryruhs 2023-10-16 15:46:29 +02:00
parent 401aa780b0
commit bc7910e79f
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
facefusion/predictor.py
View File

@ -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