Python如何實現圖片背景移除工具

這篇文章主要講解了“Python如何實現圖片背景移除工具”，文中的講解內容簡單清晰，易于學習與理解，下面請大家跟著小編的思路慢慢深入，一起來研究和學習“Python如何實現圖片背景移除工具”吧！

上傳一張圖片，實現其背景移除的功能。

代碼示例：

import streamlit as st
from rembg import remove
from PIL import Image
from io import BytesIO
import base64

st.set_page_config(layout="wide", page_title="圖片背景移除")

st.write("## 從圖片移除背景")
st.write(
    ":dog: 請上傳一張圖片，去除背景后圖片可在側邊欄下載. :grin:"
)
st.sidebar.write("## 上傳和下載 :gear:")


# Download the fixed image
def convert_image(img):
    buf = BytesIO()
    img.save(buf, format="PNG")
    byte_im = buf.getvalue()
    return byte_im


def fix_image(upload):
    image = Image.open(upload)
    col1.write("原圖片 :camera:")
    col1.image(image)

    fixed = remove(image)
    col2.write("處理后 :wrench:")
    col2.image(fixed)
    st.sidebar.markdown("\n")
    st.sidebar.download_button("下載圖片", convert_image(fixed), "fixed.png", "image/png")


col1, col2 = st.columns(2)
my_upload = st.sidebar.file_uploader("上傳圖片", type=["png", "jpg", "jpeg"])

if my_upload is not None:
    fix_image(upload=my_upload)
else:
    fix_image("./zebra.jpg")

執行

streamlit run bg_remove.py

效果圖 

感謝各位的閱讀，以上就是“Python如何實現圖片背景移除工具”的內容了，經過本文的學習后，相信大家對Python如何實現圖片背景移除工具這一問題有了更深刻的體會，具體使用情況還需要大家實踐驗證。這里是億速云，小編將為大家推送更多相關知識點的文章，歡迎關注！