TheKnight115 commited on
Commit
2b848f3
·
verified ·
1 Parent(s): 775600d

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -0
app.py ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from ultralytics import YOLO
2
+ model = YOLO("yolov8m.pt")
3
+
4
+ cap = cv2.VideoCapture(tfile.name)
5
+ while cap.isOpened():
6
+ ret, frame = cap.read()
7
+ if not ret:
8
+ break
9
+ results = model(frame)
10
+ annotated_frame = results[0].plot()
11
+ stframe.image(annotated_frame, channels="BGR")