Opencv——RGB图像的颜色分割

利用Opencv做颜色分割也是突发奇想，虽然比较简单，但是还是把代码放上来看一下，对于色彩较多的图像可能用颜色来做分割比较困难，但不排除这可以应用在一些特殊情况下，比如说…X光射线图。
代码放上来：

import cv2
import numpy as np

img = cv2.imread('test.jpg')
img_red = np.zeros(img.shape, np.uint8)

height = img.shape[0]
width = img.shape[1]

print ("img.shape[1]:%d,img.shape[0]:%d" %(img.shape[1],img.shape[0]))

for i in range(height):
        for j in range(width):
            (b,g,r) =  img[i,j]
            img_red[i,j] = (255,255,255)
            if (r>b and r>g):               #我们假设(b,g,r)中r最大时为红色，当然也可设阈值
                img_red[i,j] = (b,g,r)      #cv2中读取RGB通道的顺序是b-g-r      
                
cv2.imwrite('bp_red.jpg', img_red)
cv2.imshow('image',img)
cv2.imshow('image_red',img_red)
cv2.waitKey(0)

然后，run，再然后，结果图放上来：

看起来效果还算不错，虽然应用范围比较窄，但是也算是术业有专攻。我们用颜色分割出来后如果想进行补全的话提供思路，比如说用GAN补全；如果不想用窗口显示的话可以参考我的博客：Opencv——RGB图像的通道拆分(cv2.split)与合并(cv2.merge)，里面有opencv和matplotlib相结合的图片显示仅供参考。
感谢小伙伴的支持，觉得有帮助的话请双击666，走一波关注，谢谢大家！