Обнаружение линии по цвету с использованием Opencv в python

Question

У меня есть изображение (основное изображение графика XY, где линия сюжета имеет синий цвет, а ось x, y - в черном цвете), где мне нужно определить грани, основанные по цвету. Я натолкнулся на код ниже, где он обнаружил все строки, используя определение canny edge и алгоритм hough. Но мне нужно обнаружить только синюю линию цвета на этом изображении. Что я могу сделать, чтобы это обнаружить?

Ниже приведен код, который я использовал.

import numpy as np 
import cv2 

img = cv2.imread('xyplot.png') 
gray = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY) 
cv2.imshow('grayimage',gray) 
edges = cv2.Canny(gray,50,150,apertureSize = 3) 
cv2.imshow('edgesimage',edges) 
print img.shape[1] 
print img.shape 
minLineLength=img.shape[1]-300 
lines = cv2.HoughLinesP(image=edges,rho=0.02,theta=np.pi/500,  
    threshold=10,lines=np.array([]),         
    minLineLength=minLineLength, maxLineGap=100) 
a,b,c = lines.shape 

for i in range(a): 
    cv2.line(img, (lines[i][0][0], lines[i][0][1]), (lines[i][0][2], 
     lines[i][0] [3]), (0, 0, 255), 3, cv2.LINE_AA) 

cv2.imshow('edges', edges) 
cv2.imshow('result', img) 
cv2.waitKey(0) 
cv2.destroyAllWindows() 

Answer 1

Нанести (синий) цвет маску на изображение перед преобразованием в оттенки серого. Вы можете сделать это, указав нижнюю и верхнюю границу всех трех каналов и выполнив BITWISE_AND на исходном изображении. Вам нужно будет играть со значениями диапазонов каналов, чтобы убедиться, что только захваченные вами пиксели.

lower = np.array([200, 20, 20], dtype = "uint8") 
upper = np.array([255, 100, 100], dtype = "uint8") 
mask = cv2.inRange(img, lower, upper) 
img = cv2.bitwise_and(img, img, mask = mask) 

Примечание: Три канала в списке - B, G и R соответственно.