| import gradio as gr | |
| import numpy as np | |
| import cv2 | |
| from PIL import Image, ImageOps | |
| def filter_img(input_img,filter_type): | |
| img = input_img | |
| new_height = 1080 | |
| ratio = float(new_height) / img.shape[0] | |
| new_width = int(img.shape[1] * ratio) | |
| img = cv2.resize(img, (new_width,new_height)) | |
| filtered_img = np.zeros_like(img) | |
| 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 = np.dot(img, sepia_filter.T) | |
| filtered_img = (filtered_img / filtered_img.max() * 255).astype(np.uint8) | |
| 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(),gr.inputs.Dropdown(["Gray","B&W","Sharp","Gray-sharp","Vintage","Magenta","Yellow","Cyan","Blue","Green","Red"])], | |
| outputs= gr.components.Image(), | |
| title = "Image Filterz", | |
| description="Cool image filters!" | |
| ) | |
| demo.launch(inline=False) |