BotTracker.py 文件源码

def __process_image(self, image):

        # Our operations on the frame come here
        hsv = cv2.cvtColor(image, cv2.COLOR_BGR2HSV)
        hsv = cv2.medianBlur(hsv, 5)

        draw_col = (0,0,255)

        p1 = (LEFT_EDGE, BOTTOM_EDGE)
        p2 = (RIGHT_EDGE, TOP_EDGE)
        cv2.rectangle(hsv, p1, p2, draw_col) #bounding rectangle

        vert_spacing = (RIGHT_EDGE - LEFT_EDGE)/31. # vertical lines
        for i in range(1, 31): 
            x_pos = int(LEFT_EDGE + i*vert_spacing)
            p1 = (x_pos, BOTTOM_EDGE)
            p2 = (x_pos, TOP_EDGE)
            cv2.line(hsv, p1, p2, draw_col)

        horiz_spacing = (BOTTOM_EDGE - TOP_EDGE)/28. # horizontal lines
        for i in range(1, 28): 
            y_pos = int(TOP_EDGE + i*horiz_spacing)
            p1 = (LEFT_EDGE, y_pos)
            p2 = (RIGHT_EDGE, y_pos)
            cv2.line(hsv, p1, p2, draw_col)


        # cv2.imshow('Grid', hsv)
        # cv2.waitKey(1)
        # box around target pixel for testing
        # pt = (350, 600)
        # cv2.circle(hsv, pt, 3, draw_col, thickness =1)
        # print hsv[600][350]


        return hsv