diff --git a/ultrafastLaneDetector/ultrafastLaneDetector.py b/ultrafastLaneDetector/ultrafastLaneDetector.py index 41b865a..4a06083 100644 --- a/ultrafastLaneDetector/ultrafastLaneDetector.py +++ b/ultrafastLaneDetector/ultrafastLaneDetector.py @@ -161,7 +161,7 @@ def process_output(output, cfg): lanes_detected.append(False) lanes_points.append(lane_points) - return np.array(lanes_points), np.array(lanes_detected) + return lanes_points, np.array(lanes_detected) @staticmethod def draw_lanes(input_img, lanes_points, lanes_detected, cfg, draw_points=True):