nagasurendra commited on
Commit
a3f8ed4
·
verified ·
1 Parent(s): e78a806

Update core_pipeline.py

Browse files
Files changed (1) hide show
  1. core_pipeline.py +1 -1
core_pipeline.py CHANGED
@@ -6,7 +6,7 @@ from ultralytics import YOLO
6
  import matplotlib.pyplot as plt
7
 
8
  # Load YOLOv8 model (you can use your own tree-detection model)
9
- model = YOLO("yolov8n.pt") # Replace with tree-trained model
10
 
11
  def extract_frames(video_path, interval=30):
12
  cap = cv2.VideoCapture(video_path)
 
6
  import matplotlib.pyplot as plt
7
 
8
  # Load YOLOv8 model (you can use your own tree-detection model)
9
+ model = YOLO("./data/yolo11n.pt") # Replace with tree-trained model
10
 
11
  def extract_frames(video_path, interval=30):
12
  cap = cv2.VideoCapture(video_path)