OpenCV lane Detection 동영상

2021.04.22 - [분류 전체보기] - OpenCV lane_detection

OpenCV lane_detection

동영상의 프레임마다 적용시켜줄것 

cap = cv2.VideoCapture('data3/test2.mp4')
if cap.isOpened() == False :
    print( "Error opening video stream or file" )

else :
    while cap.isOpened() :
        # 사진을 1장씩 가져와서.
        ret, frame = cap.read()
        # canny
        canny_image = canny_edge(frame)
        # RoI
        roi_image = reg_of_interest(canny_image)
        # houghline
        lines = cv2.HoughLinesP(roi_image, 1, np.pi/180, 100, minLineLength=40, maxLineGap=5)
        # 라인 합치기
        average_lines = average_slope_intercept(frame, lines)
        # 라인 가져오기
        line_image = show_lines(frame, average_lines)
        # 원본과 라인 합치기 
        combine_image = cv2.addWeighted(frame, 0.8, line_image, 1, 1)


        # 제대로 사진 가져왔으면, 화면에 표시
        if ret == True :
            cv2.imshow("Frame", combine_image)

            # 키보드에서 esc키를 누르면 exit하라는 것
            if cv2.waitKey(25)&0xFF == 27 :
                break
            
        else :
            break
    cap.release()

cv2.destroyAllWindows()