nagasurendra commited on
Commit
cf8d1d5
·
verified ·
1 Parent(s): f1629d4

Update core_pipeline.py

Browse files
Files changed (1) hide show
  1. core_pipeline.py +5 -1
core_pipeline.py CHANGED
@@ -5,9 +5,13 @@ import numpy as np
5
  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("./data/yolo11n.pt") # Replace with tree-trained model
10
 
 
11
  def extract_frames(video_path, interval=30):
12
  cap = cv2.VideoCapture(video_path)
13
  frames = []
 
5
  from ultralytics import YOLO
6
  import matplotlib.pyplot as plt
7
 
8
+
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):
16
  cap = cv2.VideoCapture(video_path)
17
  frames = []