读入指定彩色图像flower1,分别读取并显示H、S、I各层,用学习过的基于RGB的方法将图像中的花朵分割出来,并将分割结果进行显示。使用Python代码实现加注释
时间: 2024-03-23 15:40:10 浏览: 52
首先,我们需要导入必要的库,包括OpenCV、numpy和matplotlib:
```python
import cv2
import numpy as np
import matplotlib.pyplot as plt
```
然后,我们读入指定的彩色图像flower1,并将其转换为HSV颜色空间:
```python
img = cv2.imread('flower1.jpg')
hsv_img = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
```
接下来,我们将H、S、I各层分别提取出来,并将其显示出来:
```python
h, s, v = cv2.split(hsv_img)
plt.subplot(131), plt.imshow(h, cmap='gray')
plt.title('Hue'), plt.xticks([]), plt.yticks([])
plt.subplot(132), plt.imshow(s, cmap='gray')
plt.title('Saturation'), plt.xticks([]), plt.yticks([])
plt.subplot(133), plt.imshow(v, cmap='gray')
plt.title('Intensity'), plt.xticks([]), plt.yticks([])
plt.show()
```
接下来,我们将使用学习过的基于RGB的方法将图像中的花朵分割出来。这里,我们需要先将图像从HSV颜色空间转换回RGB颜色空间:
```python
rgb_img = cv2.cvtColor(hsv_img, cv2.COLOR_HSV2RGB)
```
然后,我们可以使用阈值分割来提取花朵。这里,我们可以根据图像中花朵的颜色范围来设置阈值:
```python
lower = np.array([100, 50, 50])
upper = np.array([190, 255, 255])
mask = cv2.inRange(hsv_img, lower, upper)
```
最后,我们可以将分割结果进行显示:
```python
result = cv2.bitwise_and(rgb_img, rgb_img, mask=mask)
plt.imshow(result)
plt.title('Segmentation Result'), plt.xticks([]), plt.yticks([])
plt.show()
```
完整代码如下:
```python
import cv2
import numpy as np
import matplotlib.pyplot as plt
# 读入指定的彩色图像flower1,并将其转换为HSV颜色空间
img = cv2.imread('flower1.jpg')
hsv_img = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
# 分别提取H、S、I各层,并将其显示出来
h, s, v = cv2.split(hsv_img)
plt.subplot(131), plt.imshow(h, cmap='gray')
plt.title('Hue'), plt.xticks([]), plt.yticks([])
plt.subplot(132), plt.imshow(s, cmap='gray')
plt.title('Saturation'), plt.xticks([]), plt.yticks([])
plt.subplot(133), plt.imshow(v, cmap='gray')
plt.title('Intensity'), plt.xticks([]), plt.yticks([])
plt.show()
# 将图像从HSV颜色空间转换回RGB颜色空间
rgb_img = cv2.cvtColor(hsv_img, cv2.COLOR_HSV2RGB)
# 使用阈值分割来提取花朵
lower = np.array([100, 50, 50])
upper = np.array([190, 255, 255])
mask = cv2.inRange(hsv_img, lower, upper)
# 将分割结果进行显示
result = cv2.bitwise_and(rgb_img, rgb_img, mask=mask)
plt.imshow(result)
plt.title('Segmentation Result'), plt.xticks([]), plt.yticks([])
plt.show()
```
阅读全文