first

main.py

@@ -0,0 +1,25 @@
+import gradio as gr
+import cv2
+import numpy as np
+
+def enhance_image(opencv_image, alpha, beta, saturation_factor):
+    image = opencv_image
+    # 调整亮度和对比度
+    image = cv2.convertScaleAbs(image, alpha=alpha, beta=beta)
+    # 将图像转换为HSV颜色空间
+    hsv_image = cv2.cvtColor(image, cv2.COLOR_BGR2HSV)
+    # 调整饱和度
+    hsv_image[:, :, 1] = np.clip(hsv_image[:, :, 1] * saturation_factor, 0, 255).astype(np.uint8)
+    # 将图像转换回BGR颜色空间
+    output_image = cv2.cvtColor(hsv_image, cv2.COLOR_HSV2BGR)
+    return output_image
+
+iface = gr.Interface(
+    fn=enhance_image,
+    inputs=["image", gr.inputs.Slider(minimum=0, maximum=2, default=1.02, label="Alpha(对比度增加的比例因子)"),
+            gr.inputs.Slider(minimum=0, maximum=30, default=2, label="Beta(亮度增加的偏移量)"),
+            gr.inputs.Slider(minimum=0, maximum=2, default=1.16, label="Saturation Factor(饱和度调整因子，大于1增加饱和度，小于1减少饱和度)")],
+    outputs="image"
+)
+
+iface.launch()