|
import gradio as gr |
|
import numpy as np |
|
import cv2 |
|
|
|
def filter_img(input_img,filter_type): |
|
print('File name is : ',input_img) |
|
|
|
|
|
img = cv2.imread(input_img) |
|
|
|
filtered_img = np.zeros(img.shape, dtype=int) |
|
|
|
if filter_type == 'Vintage': |
|
sepia_filter = np.array( |
|
[[0.393, 0.769, 0.189], [0.349, 0.686, 0.168], [0.272, 0.534, 0.131]] |
|
) |
|
filtered_img = img.dot(sepia_filter.T) |
|
filtered_img /= filtered_img.max() |
|
elif filter_type == 'Gray-sharp': |
|
img = cv2.cvtColor(img,cv2.COLOR_RGB2GRAY) |
|
filtered_img = cv2.addWeighted(img,4, cv2.blur(img,(128,128)),-4,128) |
|
elif filter_type == 'Sharp': |
|
filtered_img = cv2.addWeighted(img,4, cv2.blur(img,(128,128)),-4,128) |
|
elif filter_type == 'Gray': |
|
filtered_img = cv2.cvtColor(img,cv2.COLOR_RGB2GRAY) |
|
elif filter_type == 'B&W': |
|
filtered_img = cv2.cvtColor(img,cv2.COLOR_RGB2GRAY) |
|
_,filtered_img = cv2.threshold(filtered_img,127,255,cv2.THRESH_BINARY) |
|
elif filter_type == 'Magenta': |
|
filtered_img[:,:,0] = img[:,:,0] |
|
filtered_img[:,:,2] = img[:,:,2] |
|
elif filter_type == 'Cyan': |
|
filtered_img[:,:,1] = img[:,:,1] |
|
filtered_img[:,:,2] = img[:,:,2] |
|
elif filter_type == 'Yellow': |
|
filtered_img[:,:,0] = img[:,:,0] |
|
filtered_img[:,:,1] = img[:,:,1] |
|
elif filter_type == 'Blue': |
|
filtered_img[:,:,2] = img[:,:,2] |
|
elif filter_type == 'Green': |
|
filtered_img[:,:,1] = img[:,:,1] |
|
elif filter_type == 'Red': |
|
filtered_img[:,:,0] = img[:,:,0] |
|
|
|
return filtered_img |
|
|
|
demo = gr.Interface(filter_img, |
|
inputs = [ gr.inputs.Image(type="filepath"),gr.inputs.Dropdown(["Gray","B&W","Sharp","Gray-sharp","Vintage","Magenta","Yellow","Cyan","Blue","Green","Red"])], |
|
outputs = gr.outputs.Image(type="pil"), |
|
title = "Image Filterz", |
|
description="Cool image filters!" |
|
) |
|
|
|
demo.launch(inline=False) |
