Sanjayraju30 commited on
Commit
be766da
·
verified ·
1 Parent(s): 825cd55

Update services/video_service.py

Browse files
Files changed (1) hide show
  1. services/video_service.py +7 -3
services/video_service.py CHANGED
@@ -1,6 +1,10 @@
1
- def get_live_frame(cap):
2
- while True:
 
 
 
3
  ret, frame = cap.read()
4
  if not ret:
5
- break
 
6
  yield frame
 
1
+ import cv2
2
+
3
+ def get_video_frame(video_path):
4
+ cap = cv2.VideoCapture(video_path)
5
+ while cap.isOpened():
6
  ret, frame = cap.read()
7
  if not ret:
8
+ cap.set(cv2.CAP_PROP_POS_FRAMES, 0) # Loop video
9
+ continue
10
  yield frame