OpenCV - OBJECT DETECTION

'''OBJECT DETECTION 

개념 
-입력 영상에서 사용자가 원하는 객체를 찾고 객체의 위치와 바운딩 박스를 구하는 것 

방법 

-템플릿 매칭 
-특징점과 디스크립터를 이용한 매칭 
-Cascade classifier를 이용한 검출 
-Deep learning 

--> R-CNN 계열 
--> YOLO,SSD 
--> Tensorflow Object Detection API

''''''
차선 검출 

차선 유지 보조 시스템 

-자동차가 주행 중인 차로를 벗어났을 때 운전자에게 경고 하고 주행차로로 복귀하는 제어 장치 
-지금까지 배운 기술을 활용하여 주행 중 차선만 검출 할 수 있게 구현 

 

image = cv2.imread(path + 'lane.jpg')
def reg_of_interest(image) :
    image_height = image.shape[0]
    polygons = np.array( [[ (200, image_height) , (1100, image_height), (550, 250) ]] )
    image_mask = np.zeros_like(image)
    cv2.fillPoly(image_mask, polygons, 255)
    masking_image = cv2.bitwise_and(image, image_mask)
    return masking_image

# 6. 케니에지 처리하는 함수
def canny_edge(image) :
    gray_conversion = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
    blur_conversion = cv2.GaussianBlur(gray_conversion, (5,5), 0)
    canny_conversion = cv2.Canny(blur_conversion, 50, 150)
    return canny_conversion

def show_lines(image, lines) : 
    lines_image = np.zeros_like(image)
    if lines is not None :
        for i in range(len(lines)):
            for x1,y1,x2,y2 in lines[i]:
                cv2.line(lines_image,(x1,y1),(x2,y2),(0,0,255), 10 )
    return lines_image

# 여러 선을, 하나의 선으로 만들어 주는 함수
# 방법은? 기울기와 y절편을 평균으로 해서 하나의 기울기와 y절편을 갖도록 만드는 방법.
def make_coordinates(image, line_parameters):
    slope, intercept = line_parameters
    y1 = image.shape[0]
    y2 = int(y1*(3/5))
    x1 = int((y1- intercept)/slope)
    x2 = int((y2 - intercept)/slope)
    return np.array([x1, y1, x2, y2])

def average_slope_intercept(image, lines):
    left_fit = []
    right_fit = []
    for line in lines:
        x1, y1, x2, y2 = line.reshape(4)
        parameter = np.polyfit((x1, x2), (y1, y2), 1)
        slope = parameter[0]
        intercept = parameter[1]
        if slope < 0:
            left_fit.append((slope, intercept))
        else:
            right_fit.append((slope, intercept))
    left_fit_average =np.average(left_fit, axis=0)
    right_fit_average = np.average(right_fit, axis =0)
    left_line =make_coordinates(image, left_fit_average)
    right_line = make_coordinates(image, right_fit_average)

    return np.array([[left_line, right_line]])

#이미지 가져오기
image = cv2.imread(path + 'lane.jpg')
lanelines_image = image.copy()

#흰 검으로 변환해서 라인 검출함.
canny_conversion = canny_edge(lanelines_image)
roi_conversion = reg_of_interest(canny_conversion)

#라인
lines = cv2.HoughLinesP(roi_conversion, 1, np.pi/180, 150, minLineLength = 200, maxLineGap = 5)

averaged_lines = average_slope_intercept(lanelines_image, lines)

#선을 기울기 평균값으로 적용
lines_image = show_lines(lanelines_image, averaged_lines)

#원본 이미지에 라인 그리기
combine_image = cv2.addWeighted(lanelines_image, 0.8, lines_image, 1, 1)

cv2.imshow('ori', lanelines_image)
cv2.imshow("roi", lines_image)
cv2.imshow("combined", combine_image)
cv2.waitKey(0)