manikandan18ramalingam commited on
Commit
9071273
·
verified ·
1 Parent(s): a8aaffc

Update object_detection_count.py

Browse files
Files changed (1) hide show
  1. object_detection_count.py +1 -1
object_detection_count.py CHANGED
@@ -26,7 +26,7 @@ def detectObjectsAndCount(imageFile, confidence_score, class_type):
26
  model = YOLO("yolov8n.pt") # Use other variants like yolov8s.pt for better accuracy
27
 
28
  # Define a custom directory to save the results
29
- custom_save_dir = "./"
30
 
31
  custom_read_dir = "./"
32
  class_index = getKeyByValue(model.names, class_type)
 
26
  model = YOLO("yolov8n.pt") # Use other variants like yolov8s.pt for better accuracy
27
 
28
  # Define a custom directory to save the results
29
+ custom_save_dir = "/app/runs/detect/predict"
30
 
31
  custom_read_dir = "./"
32
  class_index = getKeyByValue(model.names, class_type)