nagasurendra commited on
Commit
544c4d7
·
verified ·
1 Parent(s): 831f1ef

Update core_pipeline.py

Browse files
Files changed (1) hide show
  1. core_pipeline.py +1 -1
core_pipeline.py CHANGED
@@ -9,7 +9,7 @@ import matplotlib.pyplot as plt
9
 
10
 
11
  # Load YOLOv8 model (you can replace with your custom tree model)
12
- model = YOLO("./data/yolo11n.pt") # Replace with tree-trained model
13
 
14
 
15
  def extract_frames(video_path, interval=30):
 
9
 
10
 
11
  # Load YOLOv8 model (you can replace with your custom tree model)
12
+ model = YOLO("./data/best.pt") # Replace with tree-trained model
13
 
14
 
15
  def extract_frames(video_path, interval=30):