Spaces:
Sleeping
Sleeping
Update utils/video/frame_reader.py
Browse files
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 |
-
|
|
|
|
|
|
|
| 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 |
|