|
import gradio as gr |
|
import cv2 |
|
import numpy as np |
|
from PIL import Image, ImageEnhance |
|
from gradio_imageslider import ImageSlider |
|
|
|
def apply_filter(image, filter_type, intensity): |
|
|
|
image = np.array(image) |
|
|
|
|
|
normalized_intensity = intensity / 100.0 |
|
|
|
if filter_type == "Grayscale": |
|
return convert_to_grayscale(image) |
|
elif filter_type == "Soft Glow": |
|
base_intensity = 0.1 |
|
adjusted_intensity = base_intensity + (normalized_intensity * (1 - base_intensity)) |
|
|
|
gaussian = cv2.GaussianBlur(image, (15, 15), 0) |
|
soft_glow = cv2.addWeighted(image, 1 - adjusted_intensity, gaussian, adjusted_intensity, 0) |
|
return soft_glow |
|
elif filter_type == "Portrait Enhancer": |
|
base_intensity = 0.5 |
|
adjusted_intensity = base_intensity + (normalized_intensity * (1 - base_intensity)) |
|
|
|
image_pil = Image.fromarray(image) |
|
|
|
enhancer = ImageEnhance.Sharpness(image_pil) |
|
image_pil = enhancer.enhance(1 + 0.5 * adjusted_intensity) |
|
|
|
enhancer = ImageEnhance.Color(image_pil) |
|
image_pil = enhancer.enhance(1 + 0.5 * adjusted_intensity) |
|
|
|
enhanced_image = np.array(image_pil) |
|
return enhanced_image |
|
elif filter_type == "Warm Tone": |
|
warm_image = cv2.applyColorMap(image, cv2.COLORMAP_AUTUMN) |
|
return warm_image |
|
elif filter_type == "Cold Tone": |
|
cold_image = cv2.applyColorMap(image, cv2.COLORMAP_WINTER) |
|
return cold_image |
|
elif filter_type == "High-Key": |
|
high_key = cv2.convertScaleAbs(image, alpha=1.0 + 0.8 * normalized_intensity, beta=30) |
|
return high_key |
|
elif filter_type == "Low-Key": |
|
low_key = cv2.convertScaleAbs(image, alpha=1.0 - 0.7 * normalized_intensity, beta=-30) |
|
return low_key |
|
elif filter_type == "Haze": |
|
haze = cv2.addWeighted(image, 1.0 - 0.7 * normalized_intensity, np.full(image.shape, 255, dtype=np.uint8), 0.3 * normalized_intensity, 0) |
|
return haze |
|
else: |
|
return image |
|
|
|
def convert_to_grayscale(image): |
|
gray_image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY) |
|
return cv2.cvtColor(gray_image, cv2.COLOR_GRAY2BGR) |
|
|
|
def convert_and_save(image, filter_type, intensity): |
|
image_cv = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2BGR) |
|
filtered_image = apply_filter(image_cv, filter_type, intensity) |
|
|
|
|
|
original_image_pil = Image.fromarray(cv2.cvtColor(image_cv, cv2.COLOR_BGR2RGB)) |
|
filtered_image_pil = Image.fromarray(cv2.cvtColor(filtered_image, cv2.COLOR_BGR2RGB)) |
|
|
|
return original_image_pil, filtered_image_pil |
|
|
|
def get_filter_description(filter_type): |
|
descriptions = { |
|
"Grayscale": "์ด๋ฏธ์ง๋ฅผ ํ๋ฐฑ์ผ๋ก ๋ณํํฉ๋๋ค.", |
|
"Soft Glow": "๋ถ๋๋ฌ์ด ๋น์ ์ถ๊ฐํ์ฌ ์ด๋ฏธ์ง๋ฅผ ์์ํ๊ฒ ๋ง๋ญ๋๋ค.", |
|
"Portrait Enhancer": "ํผ๋ถ ํค์ ๊ท ์ผํ๊ฒ ํ๊ณ ์ ๋ช
๋๋ฅผ ์กฐ์ ํ์ฌ ์ธ๋ฌผ์ ๋์ฑ ๋๋ณด์ด๊ฒ ๋ง๋ญ๋๋ค.", |
|
"Warm Tone": "๋ฐ๋ปํ ์์กฐ๋ฅผ ์ถ๊ฐํ์ฌ ์ด๋ฏธ์ง์ ์จ๊ธฐ๋ฅผ ๋ํฉ๋๋ค.", |
|
"Cold Tone": "์ฐจ๊ฐ์ด ์์กฐ๋ฅผ ์ถ๊ฐํ์ฌ ์ด๋ฏธ์ง์ ์์ํจ์ ๋ํฉ๋๋ค.", |
|
"High-Key": "๋ฐ๊ณ ํ์ฌํ ์ด๋ฏธ์ง๋ฅผ ๋ง๋ค์ด๋
๋๋ค.", |
|
"Low-Key": "์ด๋์ด ํค์ ๊ฐ์กฐํ์ฌ ๋ถ์๊ธฐ ์๋ ์ด๋ฏธ์ง๋ฅผ ๋ง๋ญ๋๋ค.", |
|
"Haze": "๋ถ๋๋ฝ๊ณ ํ๋ฆฟํ ํจ๊ณผ๋ฅผ ์ถ๊ฐํ์ฌ ๋ชฝํ์ ์ธ ์ด๋ฏธ์ง๋ฅผ ๋ง๋ญ๋๋ค." |
|
} |
|
return descriptions.get(filter_type, "") |
|
|
|
with gr.Blocks() as iface: |
|
with gr.Row(): |
|
with gr.Column(): |
|
image_input = gr.Image(type="pil", label="์ด๋ฏธ์ง ์
๋ก๋") |
|
filter_input = gr.Radio( |
|
["Grayscale", "Soft Glow", "Portrait Enhancer", "Warm Tone", "Cold Tone", "High-Key", "Low-Key", "Haze"], |
|
label="ํํฐ ์ ํ", |
|
value="Soft Glow" |
|
) |
|
intensity_slider = gr.Slider(1, 100, value=50, label="ํํฐ ๊ฐ๋") |
|
description_output = gr.Markdown(get_filter_description("Soft Glow")) |
|
|
|
with gr.Column(): |
|
slider_output = ImageSlider(label="Before and After", type="pil") |
|
|
|
filter_input.change(fn=get_filter_description, inputs=filter_input, outputs=description_output) |
|
|
|
process_button = gr.Button("ํํฐ ์ ์ฉ") |
|
process_button.click( |
|
fn=convert_and_save, |
|
inputs=[image_input, filter_input, intensity_slider], |
|
outputs=slider_output |
|
) |
|
|
|
iface.title = "์ธ๋ฌผ ์ฌ์ง์ ์ต์ ํ๋ ํํฐ" |
|
iface.description = "์ธ๋ฌผ ์ฌ์ง์ ์ต์ ํ๋ ๋ค์ํ ํํฐ๋ฅผ ์ ์ฉํ ์ ์์ต๋๋ค." |
|
|
|
iface.launch() |
|
|