利用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,走一波关注,谢谢大家!