色彩阀值化处理——利用openCV-python中inRange()等相关函数,通过颜色标记代号选取不同的颜色。
#标记特定颜色区域
def Mark_Color(img, flag_color):
#自定义
if flag_color ==0:
lower = np.array([24, 43, 46], dtype="uint8")
upper = np.array([43, 255, 255], dtype="uint8")
#黑色
elif flag_color ==1:
lower = np.array([0, 0, 0], dtype="uint8")
upper = np.array([180, 255, 46], dtype="uint8")
# 灰色
elif flag_color == 2:
lower = np.array([0, 0, 46], dtype="uint8")
upper = np.array([180, 43, 220], dtype="uint8")
# 白色
elif flag_color == 3:
lower = np.array([0, 0, 221], dtype="uint8")
upper = np.array([180, 30, 255], dtype="uint8")
# 红色
elif flag_color == 4:
lower = np.array([156, 43, 46], dtype="uint8")
upper = np.array([180, 255, 255], dtype="uint8")
# 红色2
elif flag_color == 5:
lower = np.array([0, 43, 46], dtype="uint8")
upper = np.array([10, 255, 255], dtype="uint8")
# 橙色
elif flag_color == 6:
lower = np.array([11, 43, 46], dtype="uint8")
upper = np.array([25, 255, 255], dtype="uint8")
# 黄色
elif flag_color == 7:
lower = np.array([26, 43, 46], dtype="uint8")
upper = np.array([34, 255, 255], dtype="uint8")
# 绿色
elif flag_color == 8:
lower = np.array([35, 43, 46], dtype="uint8")
upper = np.array([77, 255, 255], dtype="uint8")
# 青色
elif flag_color == 9:
lower = np.array([78, 43, 46], dtype="uint8")
upper = np.array([99, 255, 255], dtype="uint8")
# 蓝色
elif flag_color == 10:
lower = np.array([100, 43, 46], dtype="uint8")
upper = np.array([124, 255, 255], dtype="uint8")
# 紫色
elif flag_color == 11:
lower = np.array([125, 43, 46], dtype="uint8")
upper = np.array([155, 255, 255], dtype="uint8")
converted = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
markingMask = cv2.inRange(converted, lower, upper)
markingMask = cv2.GaussianBlur(markingMask, (5, 5), 0)
marked_img = cv2.bitwise_and(img, img, mask=markingMask)
return marked_img
博文根据颜色不同进行匹配提到多种颜色匹配的方法,请根据需要查阅。