Spaces:
Runtime error
Runtime error
File size: 3,380 Bytes
a341af5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 |
# Checkout this tutorial
# https://blog.loginradius.com/engineering/guest-post/opencv-web-app-with-streamlit/
# Online deployment:
# https://towardsdatascience.com/3-easy-ways-to-deploy-your-streamlit-web-app-online-7c88bb1024b1
# https://www.youtube.com/watch?v=4SO3CUWPYf0
# Run: streamlit run Image_Filters_Streamlit_app.py
import io
import base64
import cv2
from PIL import Image
from filters import *
# Generating a link to download a particular image file.
def get_image_download_link(img, filename, text):
buffered = io.BytesIO()
img.save(buffered, format = 'JPEG')
img_str = base64.b64encode(buffered.getvalue()).decode()
href = f'<a href="data:file/txt;base64,{img_str}" download="{filename}">{text}</a>'
return href
# Set title.
st.title('Artistic Image Filters')
# Upload image.
uploaded_file = st.file_uploader('Choose an image file:', type=['png','jpg'])
if uploaded_file is not None:
# Convert the file to an opencv image.
raw_bytes = np.asarray(bytearray(uploaded_file.read()), dtype=np.uint8)
img = cv2.imdecode(raw_bytes, cv2.IMREAD_COLOR)
input_col, output_col = st.columns(2)
with input_col:
st.header('Original')
# Display uploaded image.
st.image(img, channels='BGR', use_column_width=True)
st.header('Filter Examples:')
# Display a selection box for choosing the filter to apply.
option = st.selectbox('Select a filter:',
( 'None',
'Black and White',
'Sepia / Vintage',
'Vignette Effect',
'Pencil Sketch',
))
# Define columns for thumbnail images.
col1, col2, col3, col4 = st.columns(4)
with col1:
st.caption('Black and White')
st.image('filter_bw.jpg')
with col2:
st.caption('Sepia / Vintage')
st.image('filter_sepia.jpg')
with col3:
st.caption('Vignette Effect')
st.image('filter_vignette.jpg')
with col4:
st.caption('Pencil Sketch')
st.image('filter_pencil_sketch.jpg')
# Flag for showing output image.
output_flag = 1
# Colorspace of output image.
color = 'BGR'
# Generate filtered image based on the selected option.
if option == 'None':
# Don't show output image.
output_flag = 0
elif option == 'Black and White':
output = bw_filter(img)
color = 'GRAY'
elif option == 'Sepia / Vintage':
output = sepia(img)
elif option == 'Vignette Effect':
level = st.slider('level', 1, 5, 2)
output = vignette(img, level)
elif option == 'Pencil Sketch':
ksize = st.slider('Blur kernel size', 1, 11, 5, step=2)
output = pencil_sketch(img, ksize)
color = 'GRAY'
with output_col:
if output_flag == 1:
st.header('Output')
st.image(output, channels=color)
# fromarray convert cv2 image into PIL format for saving it using download link.
if color == 'BGR':
result = Image.fromarray(output[:,:,::-1])
else:
result = Image.fromarray(output)
# Display link.
st.markdown(get_image_download_link(result,'output.png','Download '+'Output'),
unsafe_allow_html=True)
|