2025_march17

app.py

@@ -43,6 +43,10 @@ def gen_sources(deepfake_img):
     #----------------DeepFake Face Segmentation-----------------
     ##----------------------Initialize:Face Segmentation----------------------------------
     segmenter = FaceSegmenter(threshold=0.5)
+    #Convert Pillow Image to OpenCV Image (BGR format)
+    dfcv2_image = np.array(deepfake_img)  # Convert PIL to numpy array (RGB)
+    dfcv2_image = cv2.cvtColor(dfcv2_image, cv2.COLOR_RGB2BGR)  # Convert RGB to BGR
+    deepfake_seg = segmenter.segment_face(dfcv2_image)
     # Save PIL Image to temporary file
     with tempfile.NamedTemporaryFile(suffix=".jpg", delete=False) as temp_file:
         temp_path = temp_file.name
@@ -91,10 +95,8 @@ def gen_sources(deepfake_img):
         z_seg, _, _ = model_vaq_g.encode(seg_img) 
         rec_z_img1 = model_z1(z_df) 
         rec_z_img2 = model_z2(z_seg) 
-        rec_img1 = model_vaq_f.decode(rec_z_img1)
-        rec_img2 = model_vaq_g.decode(rec_z_img2)
-        rec_img1 = rec_img1.squeeze(0)
-        rec_img2 = rec_img2.squeeze(0)
+        rec_img1 = model_vaq_f.decode(rec_z_img1).squeeze(0)
+        rec_img2 = model_vaq_g.decode(rec_z_img2).squeeze(0)
         rec_img1_pil = T.ToPILImage()(rec_img1)
         rec_img2_pil = T.ToPILImage()(rec_img2)