✨博客主页:blog.csdn.net/wangzirui32
💖 喜欢的可以 点赞 收藏 关注哦~~
👏 我的第151篇原创作品
😎 hello,大家好,我是wangzirui32,今天我们来学习如何快速简单实现图像背景更换,开始学习吧!
1. 开发准备
Python 3
解释器OpenCV-Python
库cvzone
库mediapipe
库(cvzone
库依赖)
pip
安装命令如下:
pip install cvzone mediapipe OpenCV-Python
- 1
安装无报错后,即可开始编写代码。
2. 代码
代码如下:
import cv2
from cvzone.SelfiSegmentationModule import SelfiSegmentation
# 读取input.png
image = cv2.imread('input.png')
segmentor = SelfiSegmentation()
# 更换背景
new_image = segmentor.removeBG(image, (255, 255, 255), threshold=0.3)
# 输出照片
cv2.imwrite("output.png", new_image)
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
removeBG
参数如下:
img
需要更换背景的图片imgBg
图片背景,可以为图片对象,也可以为元组RGB值threshold
图片分割阈值,用来控制删除背景区域的大小
3. 效果
效果如下:(左为input.png
,右为output.png
)
🎉🎉🎉 好了,今天的课程就到这里,我是wangzirui32,喜欢的可以点个收藏和关注,我们下次再见!