hq gfpgan pinned py310

README.md

@@ -8,5 +8,5 @@ sdk_version: 6.25.0
 app_file: app.py
 pinned: false
 hardware: zero-a10g
+python_version: "3.10"
 ---
-Private FaceFusion ZeroGPU Space

app.py

@@ -1,16 +1,8 @@
 import spaces
 import gradio as gr
 import cv2
-import numpy as np
 import traceback
 
-def enhance_sharp(img):
-    # strong unsharp mask via opencv - works without gfpgan
-    gauss = cv2.GaussianBlur(img, (0,0), 2)
-    sharp = cv2.addWeighted(img, 1.7, gauss, -0.7, 0)
-    # extra detail enhance on face region only is done by caller
-    return sharp
-
 @spaces.GPU
 def swap_face(source_img, target_img):
     import insightface
@@ -40,35 +32,29 @@ def swap_face(source_img, target_img):
             cv2.imwrite(out,tgt)
             return out
         res = swapper.get(tgt, t_faces[0], s_faces[0], paste_back=True)
-        # enhance face region only for sharpness
+        # High quality GFPGAN enhancer
         try:
-            # detect face bbox and enhance that region
-            x1,y1,x2,y2 = map(int, t_faces[0].bbox)
-            pad = 20
-            x1,y1 = max(0,x1-pad), max(0,y1-pad)
-            x2,y2 = min(res.shape[1], x2+pad), min(res.shape[0], y2+pad)
-            face_crop = res[y1:y2, x1:x2]
-            enhanced = enhance_sharp(face_crop)
-            res[y1:y2, x1:x2] = enhanced
-        except:
-            res = enhance_sharp(res)
-        out="/tmp/swapped_enhanced.jpg"
+            from gfpgan import GFPGANer
+            enhancer = GFPGANer(model_path='https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.3.pth', upscale=1, arch='clean', channel_multiplier=2, bg_upsampler=None)
+            _, _, res = enhancer.enhance(res, has_aligned=False, only_center_face=False, paste_back=True)
+        except Exception as e:
+            print("gfpgan fail",e, flush=True)
+        out="/tmp/swapped_gfpgan.jpg"
         cv2.imwrite(out,res)
         return out
     except Exception as e:
-        tb=traceback.format_exc()[:500]
         tgt2=cv2.imread(target_img)
-        cv2.putText(tgt2, str(e)[:50], (20,40), cv2.FONT_HERSHEY_SIMPLEX, 0.6, (0,0,255),2)
+        cv2.putText(tgt2, str(e)[:60], (20,40), cv2.FONT_HERSHEY_SIMPLEX, 0.6, (0,0,255),2)
         out="/tmp/err.jpg"
         cv2.imwrite(out,tgt2)
         return out
 
 with gr.Blocks() as demo:
-    gr.Markdown("# FaceFusion Zero - Enhanced (opencv)")
+    gr.Markdown("# FaceFusion Zero + GFPGAN HQ")
     with gr.Row():
         src = gr.Image(type="filepath", label="Source")
         tgt = gr.Image(type="filepath", label="Target")
-    out = gr.Image(type="filepath", label="Enhanced Result")
-    btn = gr.Button("Swap + Enhance")
+    out = gr.Image(type="filepath", label="GFPGAN Enhanced")
+    btn = gr.Button("Swap + GFPGAN")
     btn.click(swap_face, inputs=[src, tgt], outputs=out)
 demo.launch()

requirements.txt

@@ -1,7 +1,11 @@
 gradio
 spaces
-insightface
+insightface==0.7.3
 onnxruntime-gpu
-opencv-python
+opencv-python==4.8.0.76
+gfpgan==1.3.8
+basicsr==1.4.2
+facexlib==0.3.0
+realesrgan==0.3.0
 huggingface_hub
 hf_transfer