update yolo

rohan-pandeyy · Apr 18, 2024 · 583e78e · 583e78e
1 parent ba230f9
commit 583e78e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/yolov5/combined.py b/yolov5/combined.py
@@ -172,8 +172,8 @@ def process_image_lane(image):
 dist = dist_pickle["dist"]
 
 def process_image(image):
-    lane_image = process_image_lane(image) 
-    results = model(torch.from_numpy(image).unsqueeze(0))
+    lane_image = process_image_lane(image)
+    results = model(torch.from_numpy(image).unsqueeze(0).float())
     labels, cord = results.xyxyn[0][:, -1], results.xyxyn[0][:, :-1]
     n = len(labels)
     x_shape, y_shape = image.shape[1], image.shape[0]