rengjey commited on
Commit
a144977
·
1 Parent(s): bdf3244

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,14 +6,14 @@ import torch
6
  import numpy as np
7
  from ultralytics import YOLO
8
 
9
- model = torch.hub.load('ultralytics/yolov5', 'yolov5l', pretrained=True)
10
 
11
  area = [(48,430), (18, 515), (407,485), (750,425), (690,370)]
12
  total_space = 12
13
  count=0
14
 
15
  def show_preds_video():
16
- cap = cv2.VideoCapture('V1.mp4')
17
  count=0
18
  while(cap.isOpened()):
19
  ret, frame = cap.read()
 
6
  import numpy as np
7
  from ultralytics import YOLO
8
 
9
+ model = torch.hub.load('ultralytics/yolov5', 'hi', pretrained=True)
10
 
11
  area = [(48,430), (18, 515), (407,485), (750,425), (690,370)]
12
  total_space = 12
13
  count=0
14
 
15
  def show_preds_video():
16
+ cap = cv2.VideoCapture('V111.mp4')
17
  count=0
18
  while(cap.isOpened()):
19
  ret, frame = cap.read()