KillD00zer commited on
Commit
70efac6
·
verified ·
1 Parent(s): 6b881e8

Update objec_detect_yolo.py

Browse files
Files changed (1) hide show
  1. objec_detect_yolo.py +1 -1
objec_detect_yolo.py CHANGED
@@ -28,7 +28,7 @@ def detection(path: str) -> Tuple[Set[str], str]:
28
 
29
  # Initialize YOLOv8 model with pretrained weights
30
  # Model is trained to detect: ['Fire', 'Gun', 'License_Plate', 'Smoke', 'knife']
31
- model = YOLO("yolo", "best.pt"))
32
  class_names = model.names # Get class label mappings
33
 
34
  # Set up output paths:
 
28
 
29
  # Initialize YOLOv8 model with pretrained weights
30
  # Model is trained to detect: ['Fire', 'Gun', 'License_Plate', 'Smoke', 'knife']
31
+ model = YOLO("yolo"/"best.pt"))
32
  class_names = model.names # Get class label mappings
33
 
34
  # Set up output paths: