face-correction / app.py
soiz's picture
Create app.py
e38d403 verified
raw
history blame
787 Bytes
import cv2
import gradio as gr
import numpy as np
# 顔の補正関数
def correct_face(image):
# 画像をグレースケールに変換
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
# ヒストグラム均等化で明るさ補正
corrected_image = cv2.equalizeHist(gray)
# グレースケール画像をカラー画像に変換
corrected_image = cv2.cvtColor(corrected_image, cv2.COLOR_GRAY2BGR)
return corrected_image
# Gradioインターフェースの作成
iface = gr.Interface(
fn=correct_face,
inputs=gr.Image(type="numpy"),
outputs=gr.Image(type="numpy"),
title="顔補正アプリ",
description="画像をアップロードして顔の明るさ補正を行います。"
)
# アプリを実行
iface.launch()