Update traffic_sign_detection.py

traffic_sign_detection.py

@@ -5,6 +5,7 @@ import os
 import shutil
 import cv2
 import subprocess
+from pathlib import Path
 
 # Method to detect objects with given confidence score
 
@@ -33,7 +34,8 @@ def detectTrafficObjects(imageFile, confidence_score):
         for box in result.boxes:
             print(f"Class: {model.names[int(box.cls)]}, Confidence: {box.conf}, Coordinates: {box.xyxy}")
     output_path = f"{current_path}/output"
-    #shutil.move(results[0].save_path, output_path)    
+    #shutil.move(results[0].save_path, output_path)
+    
     # Path to the saved image in the custom directory
     image_path = f"{custom_save_dir}/{imageFile}"