nullHawk commited on
Commit
d37ab35
·
verified ·
1 Parent(s): 3b6f19d

updated main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -9,7 +9,7 @@ from datetime import timedelta
9
  VIDEOS_DIR = '.'
10
  video_path = os.path.join(VIDEOS_DIR, 'sample_video.mp4')
11
  output_json_path = 'output.json'
12
- model_path = os.path.join('.', 'runs', 'detect', 'train', 'weights', 'best.pt')
13
 
14
  # Load YOLOv8 model
15
  model = YOLO(model_path) # Load a custom model
 
9
  VIDEOS_DIR = '.'
10
  video_path = os.path.join(VIDEOS_DIR, 'sample_video.mp4')
11
  output_json_path = 'output.json'
12
+ model_path = os.path.join('.', 'best.pt')
13
 
14
  # Load YOLOv8 model
15
  model = YOLO(model_path) # Load a custom model