用Python构建GUI应用的铅笔草图

在这篇文章中，我们将介绍如何将图像转换为水彩素描和铅笔素描Linux。

许多学科的艺术家都使用素描和绘画来保存想法、回忆和思想。从绘画和雕塑到参观艺术博物馆，体验艺术提供了各种健康优势，包括减少压力和提高批判性思维能力。特别是绘画、素描和油画，与提高创造力、记忆力和减轻压力有关，并被用于艺术治疗中。

通过这篇文章，我们现在可以建立一个网络应用，使用python框架Streamlit直接将图像转换成草图。用户可以上传一张图片，将其转换为水彩素描或铅笔素描。用户可以进一步下载转换后的图片，在此之前，让我们了解一些我们将在本文中使用的定义。

Streamlit – Streamlit是Python开发者中流行的开源网络应用框架。它与一系列常用的库具有互操作性和兼容性，包括Keras、Sklearn、Numpy和pandas。
PIL – PIL是Python成像库的缩写。它是一个用Python编程语言进行图像处理的软件包。它包括轻量级的图像处理工具，帮助进行图片编辑、创建和存储。
Numpy – Numpy是一个广泛使用的Python编程库，用于高级数学计算。
cv2 – 这个库被用来解决计算机视觉问题

需要的包

pip install streamlit
pip install opencv-python
pip install numpy
pip install Pillow

实现步骤

第1步：安装Streamlit

用Python构建GUI应用的铅笔草图

同样地，我们将安装PIL、Numpy和cv2。

第二步：测试安装是否成功。

streamlit hello

用Python构建GUI应用的铅笔草图

第3步：现在运行streamlit网络应用程序。我们需要键入以下命令

streamlit run app.py

用Python构建GUI应用的铅笔草图

第四步：现在，网络应用已经成功启动。你可以通过本地URL或网络URL访问该网络应用。

用Python构建GUI应用的铅笔草图

第5步：创建一个新的文件夹，命名为 – 网络应用程序，将图像转换成草图。

第6步：将网络应用程序的代码粘贴在文件’app.py‘中并保存该文件。

最初在代码中，我们导入了所有需要的框架、包、库和模块，我们将利用它们来构建网络应用。此外，我们必须使用用户定义的函数，用于将图像转换成水彩素描和将图像转换成铅笔素描。还有一个函数是利用PIL库加载图像的。主函数中有网络应用的代码。最初，我们有一些标题和副标题来引导用户上传图片。为了上传图片，我们利用了streamlit的文件上传器。我们还提供了一个下拉菜单，让用户选择制作水彩素描/制作铅笔素描，然后根据他们的选择，我们渲染结果。原始图像和应用滤镜后的图像都是并排呈现的，这样用户就可以比较这两张图像的结果。最后，用户还可以将图像下载到他们的本地机器上。这可以通过利用Streamlit的下载按钮来完成。

完整代码

# import the frameworks, packages and libraries
import streamlit as st
from PIL import Image
from io import BytesIO
import numpy as np
import cv2  # computer vision
  
# function to convert an image to a 
# water color sketch
def convertto_watercolorsketch(inp_img):
    img_1 = cv2.edgePreservingFilter(inp_img, flags=2, sigma_s=50, sigma_r=0.8)
    img_water_color = cv2.stylization(img_1, sigma_s=100, sigma_r=0.5)
    return(img_water_color)
  
# function to convert an image to a pencil sketch
def pencilsketch(inp_img):
    img_pencil_sketch, pencil_color_sketch = cv2.pencilSketch(
        inp_img, sigma_s=50, sigma_r=0.07, shade_factor=0.0825)
    return(img_pencil_sketch)
  
# function to load an image
def load_an_image(image):
    img = Image.open(image)
    return img
  
# the main function which has the code for
# the web application
def main():
    
    # basic heading and titles
    st.title('WEB APPLICATION TO CONVERT IMAGE TO SKETCH')
    st.write("This is an application developed for converting\
    your ***image*** to a ***Water Color Sketch*** OR ***Pencil Sketch***")
    st.subheader("Please Upload your image")
      
    # image file uploader
    image_file = st.file_uploader("Upload Images", type=["png", "jpg", "jpeg"])
  
    # if the image is uploaded then execute these 
    # lines of code
    if image_file is not None:
        
        # select box (drop down to choose between water 
        # color / pencil sketch)
        option = st.selectbox('How would you like to convert the image',
                              ('Convert to water color sketch',
                               'Convert to pencil sketch'))
        if option == 'Convert to water color sketch':
            image = Image.open(image_file)
            final_sketch = convertto_watercolorsketch(np.array(image))
            im_pil = Image.fromarray(final_sketch)
  
            # two columns to display the original image and the
            # image after applying water color sketching effect
            col1, col2 = st.columns(2)
            with col1:
                st.header("Original Image")
                st.image(load_an_image(image_file), width=250)
  
            with col2:
                st.header("Water Color Sketch")
                st.image(im_pil, width=250)
                buf = BytesIO()
                img = im_pil
                img.save(buf, format="JPEG")
                byte_im = buf.getvalue()
                st.download_button(
                    label="Download image",
                    data=byte_im,
                    file_name="watercolorsketch.png",
                    mime="image/png"
                )
  
        if option == 'Convert to pencil sketch':
            image = Image.open(image_file)
            final_sketch = pencilsketch(np.array(image))
            im_pil = Image.fromarray(final_sketch)
              
            # two columns to display the original image
            # and the image after applying
            # pencil sketching effect
            col1, col2 = st.columns(2)
            with col1:
                st.header("Original Image")
                st.image(load_an_image(image_file), width=250)
  
            with col2:
                st.header("Pencil Sketch")
                st.image(im_pil, width=250)
                buf = BytesIO()
                img = im_pil
                img.save(buf, format="JPEG")
                byte_im = buf.getvalue()
                st.download_button(
                    label="Download image",
                    data=byte_im,
                    file_name="watercolorsketch.png",
                    mime="image/png"
                )
  
  
if __name__ == '__main__':
    main()