IceKhoffi commited on
Commit
5adab85
·
verified ·
1 Parent(s): d964c35

Update utils/video/frame_reader.py

Browse files
Files changed (1) hide show
  1. utils/video/frame_reader.py +3 -1
utils/video/frame_reader.py CHANGED
@@ -78,7 +78,9 @@ class FrameReader(threading.Thread):
78
  ret, frame = self.cap.read()
79
  if not ret:
80
  if os.path.exists(self.video_url):
81
- break
 
 
82
  time.sleep(0.5)
83
  continue
84
 
 
78
  ret, frame = self.cap.read()
79
  if not ret:
80
  if os.path.exists(self.video_url):
81
+ print(f"[FrameReader] End of file reached. Looping {self.video_url}")
82
+ self.cap.set(cv2.CAP_PROP_POS_FRAMES, 0)
83
+ continue
84
  time.sleep(0.5)
85
  continue
86