Color Space Conversion的代码注释如下:
import cv2 as cv
src = cv.imread("dataset/train/bees/1.png")
cv.namedWindow("rgb", cv.WINDOW_AUTOSIZE)
cv.imshow("rgb", src)
# RGB to HSV
hsv = cv.cvtColor(src, cv.COLOR_BGR2HSV)
cv.imshow("hsv", hsv)
# RGB to YUV
yuv = cv.cvtColor(src, cv.COLOR_BGR2YUV)
cv.imshow("yuv", yuv)
# RGB to YCRCB
ycrcb = cv.cvtColor(src, cv.COLOR_BGR2YCrCb)
cv.imshow("ycrcb", ycrcb)
#获取图像MASK
src2 = cv.imread("dataset/train/3.jpg");
cv.imshow("src2", src2)
hsv = cv.cvtColor(src2, cv.COLOR_BGR2HSV)
mask = cv.inRange(hsv, (35, 43, 46), (99, 255, 255))
cv.imshow("mask", mask)
cv.waitKey(0)
cv.destroyAllWindows()
运行结果如图:
总结:
一、
此代码的上半部分是把RGB与YUV、HSV、YCRCB颜色空间模型的转换。至于有什么区别和转换原理看不懂,可以百度或者看这个博客,感觉就是图片的显示方法不同而已。例如HSV介绍:
1.HSV是一种比较直观的颜色模型,所以在许多图像编辑工具中应用比较广泛,这个模型中颜色的参数分别是:色调(H, Hue),饱和度(S,Saturation),明度(V, Value)。
2.利用openCV中cvSplit函数的在选择图像IPL_DEPTH_32F类型时,H取值范围是0-360,S取值范围是0-1(0%-100%),V取值范围是0-1(0%-100%)。
3.利用openCV中cvSplit函数的在选择图像IPL_DEPTH_8UC类型时,H取值范围是0-180,S取值范围是0-255,V取值范围是0-255。
二、
此代码的下部分是获取图像mask,是OpenCV数字图像处理之ROI区域的提取的一个步骤。cv.inRange(hsv, (h_min, s_min, v_min), (h_max, s_max, v_max))函数cv.inRange函数通过设置不同的h、s、v的min和max阈值可以获取不同色彩的一个二值的mask图,下图为各颜色的阈值表:
OpenCV数字图像处理之ROI区域的提取可以把结果中人物的绿色背景改为其他颜色,具体步骤是:
先把人物的mask提取出来
获取人物:
新建一张与原始图一样大小的蓝色的背景图,并得到蓝色背景的mask
将人物图贴到蓝色背景上:
具体操作参考此代码