forked from scott/YOLOv8_test_example
parse image path from argument
This commit is contained in:
1 parent
a9a3f2142b
commit
7df0ad92ae
1 file changed
+8
-2
@@ -1,10 +1,16 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
|
||||||
|
import sys
|
||||||
|
|
||||||
from ultralytics import YOLO
|
from ultralytics import YOLO
|
||||||
|
|
||||||
|
image_path = sys.argv[1]
|
||||||
|
|
||||||
# Load the small pretrained YOLOv8 object-detection model.
|
# Load the small pretrained YOLOv8 object-detection model.
|
||||||
model = YOLO("yolov8n.pt")
|
model = YOLO("yolov8n.pt")
|
||||||
|
|
||||||
# Run object detection on a test image.
|
# Run object detection on a test image.
|
||||||
results = model("bus4.jpg")
|
results = model(image_path)
|
||||||
|
|
||||||
has_bus = False
|
has_bus = False
|
||||||
max_confidence = 0.0
|
max_confidence = 0.0
|
||||||
@@ -13,7 +19,7 @@ for result in results:
|
|||||||
class_names = result.names
|
class_names = result.names
|
||||||
detected_classes = result.boxes.cls.tolist()
|
detected_classes = result.boxes.cls.tolist()
|
||||||
confidences = result.boxes.conf.tolist()
|
confidences = result.boxes.conf.tolist()
|
||||||
|
|
||||||
# Check each detected object and its matching confidence score
|
# Check each detected object and its matching confidence score
|
||||||
for cls_id, conf in zip(detected_classes, confidences):
|
for cls_id, conf in zip(detected_classes, confidences):
|
||||||
if class_names[int(cls_id)] == "bus":
|
if class_names[int(cls_id)] == "bus":
|
||||||
|
|||||||
Reference in new issue
Block a user