search

ibaiGorordo / TfLite-Ultra-Fast-Lane-Detection-Inference

Example scripts for the detection of lanes using the ultra fast lane detection model in Tensorflow Lite.
MIT License
21 stars 2 forks source link

Error encountered when running inference on raspberry pi and edge TPU #6

Open vmelinawo opened 2 months ago

vmelinawo commented 2 months ago

I keep encountering this error from the ultraFastLaneDetector.py script:

Traceback (most recent call last): File "/home/MENG/Downloads/ultra/videoLaneDetection.py", line 31, in output_img = lane_detector.detect_lanes(frame) File "/home/MENG/Downloads/ultra/ultrafastLaneDetector/ultrafastLaneDetector.py", line 84, in detect_lanes self.lanes_points, self.lanes_detected = self.process_output(output, self.cfg) File "/home/MENG/Downloads/ultra/ultrafastLaneDetector/ultrafastLaneDetector.py", line 149, in process_output loc = np.sum(prob * idx, axis=0) ValueError: operands could not be broadcast together with shapes (200,18,4) (100,1,1)