import numpy as np import cv2 class ColorMeter(object): color_hsv = { # HSV,H表示色调(度数表示0-180),S表示饱和度(取值0-255),V表示亮度(取值0-255) # "orange": [np.array([11, 115, 70]), np.array([25, 255, 245])], "yellow": [np.array([11, 115, 70]), np.array([34, 255, 245])], "green": [np.array([35, 115, 70]), np.array([77, 255, 245])], "lightblue": [np.array([78, 115, 70]), np.array([99, 255, 245])], "blue": [np.array([100, 115, 70]), np.array([124, 255, 245])], "purple": [np.array([125, 115, 70]), np.array([155, 255, 245])], "red": [np.array([156, 115, 70]), np.array([179, 255, 245])], } def __init__(self, is_show=False): self.is_show = is_show self.img_shape = None def detect_color(self, frame): self.img_shape = frame.shape res = {} # 将图像转化为HSV格式 hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV) for text, range_ in self.color_hsv.items(): # 去除颜色范围外的其余颜色 mask = cv2.inRange(hsv, range_[0], range_[1]) erosion = cv2.erode(mask, np.ones((1, 1), np.uint8), iterations=2) dilation = cv2.dilate(erosion, np.ones((1, 1), np.uint8), iterations=2) target = cv2.bitwise_and(frame, frame, mask=dilation) # 将滤波后的图像变成二值图像放在binary中 ret, binary = cv2.threshold(dilation, 127, 255, cv2.THRESH_BINARY) # 在binary中发现轮廓,轮廓按照面积从小到大排列 contours, hierarchy = cv2.findContours( binary, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE ) if len(contours) > 0: # cv2.boundingRect()返回轮廓矩阵的坐标值,四个值为x, y, w, h, 其中x, y为左上角坐标,w,h为矩阵的宽和高 boxes = [ box for box in [cv2.boundingRect(c) for c in contours] if min(frame.shape[0], frame.shape[1]) / 10 < min(box[2], box[3]) < min(frame.shape[0], frame.shape[1]) / 1 ] if boxes: res[text] = boxes if self.is_show: for box in boxes: x, y, w, h = box # 绘制矩形框对轮廓进行定位 cv2.rectangle( frame, (x, y), (x + w, y + h), (153, 153, 0), 2 ) # 将绘制的图像保存并展示 # cv2.imwrite(save_image, img) cv2.putText( frame, # image text, # text (x, y), # literal direction cv2.FONT_HERSHEY_SIMPLEX, # dot font 0.9, # scale (255, 255, 0), # color 2, # border ) if self.is_show: cv2.imshow("image", frame) cv2.waitKey(1) # cv2.destroyAllWindows() return res if __name__ == "__main__": cap = cv2.VideoCapture(0) m = ColorMeter(is_show=True) while True: success, frame = cap.read() res = m.detect_color(frame) print(res) if cv2.waitKey(1) & 0xFF == ord('q'): break
时间: 2023-03-15 22:05:54 浏览: 262
"red": (0, 255, 255),
"green": (85, 255, 128),
"blue": (170, 255, 128) } 你好!我能够理解你正在询问的是如何使用HSV色彩空间来表示不同的颜色。例如,红色的HSV值为(0,255,255),绿色的HSV值为(85,255,128),蓝色的HSV值为(170,255,128)。
相关问题
逐行解释以下代码 import cv2 import numpy as np def cvtBackground(path,color): """ 功能:给证件照更换背景色(常用背景色红、白、蓝) 输入参数:path:照片路径 color:背景色 """ im=cv2.imread(path) im_hsv=cv2.cvtColor(im,cv2.COLOR_BGR2HSV) #BGR和HSV的转换使用 cv2.COLOR_BGR2HSV #aim=np.uint8([[im[0,0,:]]]) #hsv_aim=cv2.cvtColor(aim,cv2.COLOR_BGR2HSV) mask=cv2.inRange(im_hsv,np.array([im_hsv[0,0,0]-0.1,100,100]),np.array([im_hsv[0,0,0]+0.1,255,255])) #利用cv2.inRange函数设阈值,去除背景部分 mask1=mask #在lower_red~upper_red之间的值变成255 img_median = cv2.medianBlur(mask,5) #自己加,中值滤波,去除一些边缘噪点 mask2 = img_median mask_inv=cv2.bitwise_not(mask2) img1=cv2.bitwise_and(im,im,mask=mask_inv) #将人物抠出 bg=im.copy() rows,cols,channels=im.shape bg[:rows,:cols,:]=color img2=cv2.bitwise_and(bg,bg,mask=mask2) #将背景底板抠出 img=cv2.add(img1,img2) #改变图片比例 h, w = img.shape[:2] img5 = cv2.resize(img, (int(w * 1/3), int(h * 1/3)), interpolation=cv2.INTER_LINEAR) image={'im':im,'im_hsv':im_hsv,'mask':mask1,'img_median':img_median,'img':img5} cv2.startWindowThread() #加了这个后在图片窗口按Esc就可以关闭图片窗口 for key in image: cv2.namedWindow(key) cv2.imshow(key,image[key]) cv2.waitKey(0) cv2.destroyAllWindows() return image #test if __name__=='__main__': img=cvtBackground('zjz.jpeg',[55,55,55])
这段代码实现了给证件照更换背景色的功能,具体解释如下:
1. 首先导入需要用到的库:cv2 (OpenCV) 和 numpy。
2. 定义了一个名为 cvtBackground 的函数,其中 path 表示图片路径,color 表示要更换的背景颜色,可以是红、白或蓝。
3. 使用 cv2.imread 函数读取图片,并使用 cv2.COLOR_BGR2HSV 将 BGR 格式的图片转换为 HSV 格式。
4. 利用 cv2.inRange 函数设定阈值,去除背景部分,生成掩膜 mask。
5. 使用中值滤波函数 cv2.medianBlur 去除一些边缘噪点,并将处理后的掩膜保存为 mask2。
6. 利用 cv2.bitwise_not 函数生成 mask_inv,表示掩膜的反转。
7. 使用 cv2.bitwise_and 函数将人物抠出,保存为 img1。
8. 将底板 bg 生成与图片尺寸相同的背景,并将其赋值为 color。
9. 使用 cv2.bitwise_and 函数将背景底板抠出,保存为 img2。
10. 使用 cv2.add 函数将抠出的人物和背景合成一张图片,保存为 img。
11. 最后使用 cv2.resize 函数改变图片比例,将图片缩小至原图的1/3,并将处理后的各个变量以字典的形式保存到 image 中,便于后续查看处理效果。
12. 在主函数中调用 cvtBackground 函数并传入图片路径和要更换的背景颜色,将返回的 image 变量中的图像显示在窗口中。
import cv2 import numpy as np def cvtBackground(path,color): """ 功能:给证件照更换背景色(常用背景色红、白、蓝) 输入参数:path:照片路径 color:背景色 """ im=cv2.imread(path) im_hsv=cv2.cvtColor(im,cv2.COLOR_BGR2HSV) #BGR和HSV的转换使用 cv2.COLOR_BGR2HSV #aim=np.uint8([[im[0,0,:]]]) #hsv_aim=cv2.cvtColor(aim,cv2.COLOR_BGR2HSV) mask=cv2.inRange(im_hsv,np.array([im_hsv[0,0,0]-0.1,100,100]),np.array([im_hsv[0,0,0]+0.1,255,255])) #利用cv2.inRange函数设阈值,去除背景部分 mask1=mask #在lower_red~upper_red之间的值变成255 img_median = cv2.medianBlur(mask,5) #自己加,中值滤波,去除一些边缘噪点 mask2 = img_median mask_inv=cv2.bitwise_not(mask2) img1=cv2.bitwise_and(im,im,mask=mask_inv) #将人物抠出 bg=im.copy() rows,cols,channels=im.shape bg[:rows,:cols,:]=color img2=cv2.bitwise_and(bg,bg,mask=mask2) #将背景底板抠出 img=cv2.add(img1,img2) #改变图片比例 h, w = img.shape[:2] img5 = cv2.resize(img, (int(w * 1/3), int(h * 1/3)), interpolation=cv2.INTER_LINEAR) image={'im':im,'im_hsv':im_hsv,'mask':mask1,'img_median':img_median,'img':img5} cv2.startWindowThread() #加了这个后在图片窗口按Esc就可以关闭图片窗口 for key in image: cv2.namedWindow(key) cv2.imshow(key,image[key]) cv2.waitKey(0) cv2.destroyAllWindows() return image #test if __name__=='__main__': img=cvtBackground('D:\picture\\rm.jpg',[255,255,255])设计思路
这段代码的主要功能是将证件照片的背景色更换为指定的颜色(常用红、白、蓝),具体思路如下:
1. 读取照片,将BGR格式转换为HSV格式;
2. 设定阈值,将背景部分的像素点设为0,人物部分的像素点设为255,得到背景掩膜;
3. 对背景掩膜进行中值滤波,去除一些边缘噪点;
4. 利用背景掩膜和人物掩膜,将人物抠出;
5. 创建一个与原图大小相同的底板,并设定为指定的背景颜色;
6. 利用背景掩膜和底板掩膜,将底板抠出;
7. 将人物和底板拼接在一起,形成新的图片;
8. 将图片比例缩小为原来的1/3,方便观察和保存。
代码中使用的OpenCV库是一个流行的计算机视觉库,可以用于图像处理、计算机视觉、机器学习等领域。
阅读全文