这篇文章主要讲解了“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如何实现图片背景移除工具这一问题有了更深刻的体会,具体使用情况还需要大家实践验证。这里是亿速云,小编将为大家推送更多相关知识点的文章,欢迎关注!
免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。