用opencv 写一个提取图片花瓣作为前景的代码
时间: 2023-12-03 18:44:38 浏览: 181
以下是一个简单的基于 OpenCV 的 Python 代码,可以提取图片的花瓣作为前景。
首先需要安装 OpenCV 库:
```
pip install opencv-python
```
然后,将以下代码保存为 `flower_extraction.py` 文件,并将需要处理的图片命名为 `flower.jpg`,放在同一个目录下。
```python
import cv2
# 读入图片
img = cv2.imread('flower.jpg')
# 转换为 HSV 颜色空间
hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
# 设定提取的范围
lower_range = (100, 50, 50)
upper_range = (130, 255, 255)
# 提取符合范围内颜色的区域
mask = cv2.inRange(hsv, lower_range, upper_range)
# 对原图进行掩模操作,提取前景
res = cv2.bitwise_and(img, img, mask=mask)
# 显示结果
cv2.imshow('Original', img)
cv2.imshow('Mask', mask)
cv2.imshow('Result', res)
# 按下任意键退出
cv2.waitKey(0)
# 关闭窗口
cv2.destroyAllWindows()
```
运行代码后,会弹出三个窗口:原图、符合范围的颜色区域掩模、提取出的前景。其中,掩模窗口的黑色区域为非前景区域,白色区域为前景区域。可以根据需要调整提取范围和参数。
阅读全文