예제 #1
0
def test_renqder(vehicle: Vehicle) -> None:
    while True:
        vehicle.update_parts()
        gray = vehicle.mem.get(vehicle.parts[1]['outputs'])[0]
        gray2 = vehicle.mem.get(vehicle.parts[4]['outputs'])[0]
        blur = vehicle.mem.get(vehicle.parts[5]['outputs'])[0]
        canny = vehicle.mem.get(vehicle.parts[6]['outputs'])[0]
        # hough = vehicle.mem.get(vehicle.parts[6]['outputs'])[0]
        road = vehicle.mem.get(vehicle.parts[8]['outputs'])[0]
        # threshold = vehicle.mem.get(vehicle.parts[6]['outputs'])[0]
        # dilate = vehicle.mem.get(vehicle.parts[7]['outputs'])[0]
        cv2.imshow('gray', gray)
        cv2.imshow('gray2', gray2)
        cv2.imshow('blur', blur)
        cv2.imshow('canny', canny)
        # cv2.imshow('hough', hough)
        cv2.imshow('road', road)
        # cv2.imshow('threshold', threshold)
        # cv2.imshow('dilate', dilate)
        if cv2.waitKey(1) & 0xFF == ord('q'):
            break

        if cv2.waitKey(1) & 0xFF == ord(' '):
            while True:
                if cv2.waitKey(1) & 0xFF == ord(' '):
                    break
        sleep(0.01)
예제 #2
0
def test_render(vehicle: Vehicle) -> None:
    while True:
        vehicle.update_parts()
        gray = vehicle.mem.get([ConvertToGrayPart.IMG_GRAY_RAW])[0]
        threshold = vehicle.mem.get([ThresholdPart.IMG_THRESHOLD])[0]
        road = vehicle.mem.get([RoadDebugPart.IMG_ROAD])[0]
        road_ellipse = vehicle.mem.get([RoadEllipseDebugPart.IMG_ROAD_ELLIPSE])[0]
        cv2.imshow('gray', gray)
        cv2.imshow('threshold', threshold)
        cv2.imshow('road', road)
        if road_ellipse is not None:
            cv2.imshow('road_ellipse', road_ellipse)

        if cv2.waitKey(1) & 0xFF == ord('q'):
            break

        if cv2.waitKey(1) & 0xFF == ord(' '):
            while True:
                if cv2.waitKey(1) & 0xFF == ord(' '):
                    break
        sleep(0.01)