Update objec_detect_yolo.py
Browse files- 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"
|
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:
|