Python实现图片背景移除工具
更新时间：2023年02月27日 09:57:52 作者：吃肉的小馒头
这篇文章主要为大家详细介绍了如何通过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图片背景移除内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！
相关文章
本文主要介绍了python使用xpath获取页面元素的使用，文中通过示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2021-09-09
python PyQt5中QRadioButton的详细使用教程与应用实战
PyQt5是一个跨平台的GUI工具包,用于创建具有Python绑定的Qt应用程序,在PyQt5中,QRadioButton是一个非常有用的控件,用于在用户界面上提供单选选项,本文将详细介绍QRadioButton的基本用法、常用属性和方法,需要的朋友可以参考下2024-08-08
python神经网络tfrecords文件的写入读取及内容解析
这篇文章主要为大家介绍了python神经网络tfrecords文件的写入读取及内容解析，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2022-05-05
这篇文章主要给大家介绍了关于Python配置文件处理的相关资料，文中通过示例代码介绍的非常详细，对大家学习或者使用Python具有一定的参考学习价值，需要的朋友们下面来一起学习学习吧2019-08-08
在进行深度学习处理时,有些的代码处理的数据格式为npy,但是常常有的数据格式为png,因此本文就来介绍一下Python如何实现图像批量png格式转为npy格式,需要的可以参考下2023-12-12
这篇文章主要介绍了Flask搭建虚拟环境并运行第一个flask程序，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2021-04-04
在本文中，我们将学习如何使用 Brownie、Python 和 Chainlink 来制作非同质化的 NFT 作品，并在 OpenSea NFT 市场上展示和销售我们的成果。2021-06-06
这篇文章主要介绍了python记录程序运行时间的三种方法的相关资料,需要的朋友可以参考下2017-07-07
今天小编就为大家分享一篇使用python来调用CAN通讯的DLL实现方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-07-07
一步一步教你用Python pyglet仿制鸿蒙系统里的时钟
pyglet是一个面向Python的跨平台窗口、多媒体库,它可以用于创建游戏和多媒体应用程序,下面这篇文章主要给大家介绍了关于如何一步一步教你用Python pyglet仿制鸿蒙系统里的时钟,需要的朋友可以参考下2024-03-03
最新评论