Update traffic_sign_detection.py

traffic_sign_detection.py

@@ -45,12 +45,12 @@ def detectTrafficObjects(imageFile, confidence_score):
     #plt.axis("off")  # Hide axes
     #plt.show()
 
-def detectVideo(videoFile, confidence_score):
+def detectTrafficVideo(videoFile, confidence_score):
     # Load the video
     #video_path = "./Bangkok.mp4"
     cap = cv2.VideoCapture(videoFile)
     
-    model = YOLO("yolov8n.pt", verbose=False)
+    model = YOLO("./yolov8n_trained_traffic_signs.pt", verbose=False)
     # Get video properties
     fps = int(cap.get(cv2.CAP_PROP_FPS))
     width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))