예제 #1
0
def fixture_vehicle(video_camera: VideoCamera, road_config: RoadConfigController) -> Vehicle:
    vehicle = Vehicle()
    vehicle.register(video_camera)
    vehicle.register(ConvertToGrayPart())
    vehicle.register(BoundingBoxPart(input_img_key=ConvertToGrayPart.IMG_GRAY_RAW,
                                     output_img_key=ConvertToGrayPart.IMG_GRAY_RAW))
    vehicle.register(HistogramPart())
    vehicle.register(ThresholdPart())
    vehicle.register(BlurPart(input_key=ThresholdPart.IMG_THRESHOLD, output_key='img/blur'))
    vehicle.register(CannyPart(input_img_key='img/blur', output_img_key='img/canny'))
    # vehicle.register(HoughPart(input_img_key='img/canny', output_img_key='img/hough'))
    vehicle.register(RoadPart(road_config, input_img_type='img/canny'))
    vehicle.register(RoadDebugPart())

    # vehicle.register(BoundingBoxPart(input_img_key=IMG_GRAY_RAW, output_img_key=IMG_GRAY_RAW))
    # vehicle.register(HistogramPart())
    # vehicle.register(GraySelectorPart())
    # vehicle.register(ThresholdController(config=ThresholdConfigController(limit_min=180, limit_max=200,
    # threshold_default = 190,
    # threshold_delta = 10,
    # threshold_dynamic = False, mqtt_enable = False)))
    # vehicle.register(DilatePart())
    # vehicle.register(RoadPart(road_config))
    # vehicle.register(RoadDebugPart())
    return vehicle
예제 #2
0
def fixture_vehicle(video_camera: VideoCamera,
                    road_config: RoadConfigController) -> Vehicle:
    vehicle = Vehicle()
    vehicle.register(video_camera)
    vehicle.register(ConvertToGrayPart())
    vehicle.register(
        BoundingBoxPart(input_img_key=ConvertToGrayPart.IMG_GRAY_RAW,
                        output_img_key=ConvertToGrayPart.IMG_GRAY_RAW))
    vehicle.register(HistogramPart())
    vehicle.register(ThresholdPart())
    vehicle.register(
        BlurPart(input_key=ThresholdPart.IMG_THRESHOLD,
                 output_key='img/blur',
                 kernel_size=5))
    vehicle.register(
        CannyPart(input_img_key='img/blur',
                  output_img_key='img/canny',
                  low_threshold=50,
                  high_threshold=150))
    # vehicle.register(HoughPart(input_img_key='img/canny', output_img_key='img/hough'))
    vehicle.register(RoadPart(road_config, input_img_type='img/canny'))
    vehicle.register(RoadDebugPart())

    return vehicle
예제 #3
0
def fixture_vehicle(video_camera: VideoCamera,
                    road_config: RoadConfigController,
                    throttle_config: ThrottleConfigController) -> Vehicle:
    vehicle = Vehicle()
    vehicle.register(video_camera)
    vehicle.register(ComponentRoadPart())
    vehicle.register(AngleRoadPart())
    vehicle.register(
        ThrottleEllipsePart(throttle_config_controller=throttle_config))
    vehicle.register(RoadEllipseDebugPart())
    vehicle.register(
        ThrottleDebugPart(input_img_key=RoadEllipseDebugPart.IMG_ROAD_ELLIPSE))
    return vehicle