Upload feifeifilter.py

feifeilib/feifeifilter.py

@@ -0,0 +1,67 @@
+# feifeilib/feifeifilter.py
+import cv2
+import numpy as np
+from PIL import Image
+
+def validate_image(img):
+    """确保图像为3通道uint8格式"""
+    if img.dtype != np.uint8:
+        img = img.astype(np.uint8)
+    if len(img.shape) == 2:
+        img = cv2.cvtColor(img, cv2.COLOR_GRAY2BGR)
+    elif img.shape[2] == 4:
+        img = cv2.cvtColor(img, cv2.COLOR_BGRA2BGR)
+    return img
+
+def japanese_style(
+    input_img, 
+    brightness=15, 
+    blue_intensity=1.1,
+    softlight_strength=0.2,
+    sharpen_strength=9
+):
+    """日系风格处理函数（已修复参数接收问题）"""
+    # 参数类型转换（Gradio的Slider返回float类型）
+    brightness = int(brightness)
+    blue_intensity = float(blue_intensity)
+    softlight_strength = float(softlight_strength)
+    sharpen_strength = int(sharpen_strength)
+
+    # 格式转换
+    img = np.array(input_img)
+    img = cv2.cvtColor(img, cv2.COLOR_RGB2BGR)
+    img = validate_image(img)
+
+    # ===== 基础调整 =====
+    img = cv2.convertScaleAbs(img, alpha=1.1, beta=brightness)
+
+    # ===== 色调调整 =====
+    b, g, r = cv2.split(img)
+    b = np.clip(b * blue_intensity, 0, 255).astype(np.uint8)
+    g = np.clip(g * 1.05, 0, 255).astype(np.uint8)
+    img = cv2.merge((b, g, r))
+
+    # ===== 柔光效果 =====
+    overlay = img.copy()
+    cv2.rectangle(overlay, (0,0), (img.shape[1], img.shape[0]), 
+                (210,230,250), -1)
+    img = cv2.addWeighted(overlay, softlight_strength, 
+                        img, 1-softlight_strength, 0)
+
+    # ===== 曲线调整 =====
+    lut = np.array([int(255 * (0.5 * np.sin((x/255 - 0.5)*np.pi) + 0.5)) 
+                   for x in range(256)])
+    img = cv2.LUT(img, lut)
+    
+    # 锐化前再次验证格式
+    img = validate_image(img)
+
+    # ===== 锐化处理 =====
+    # kernel = np.array([[-1,-1,-1], 
+    #                  [-1,sharpen_strength,-1], 
+    #                  [-1,-1,-1]])
+    # img = cv2.filter2D(img, cv2.CV_8U, kernel)
+    
+    # 转回PIL格式
+    img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
+    return Image.fromarray(img)