mirrors/yolov5
1
0
Fork 0
mirror of https://github.com/ultralytics/yolov5.git synced 2025-06-03 14:49:29 +08:00
Code Issues Projects Releases Wiki Activity

Update LoadImages ret_val=False handling (#5852)

Browse Source 
Video errors may occur.
This commit is contained in:
gggmt 2021-12-02 17:57:39 +08:00 committed by GitHub
parent e8f8f2b903
commit 1679aacdc7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
utils/datasets.py
View File

@ -200,7 +200,7 @@ class LoadImages:
# Read video # Read video
self.mode = 'video' self.mode = 'video'
ret_val, img0 = self.cap.read() ret_val, img0 = self.cap.read()
if not ret_val: while not ret_val:
self.count += 1 self.count += 1
self.cap.release() self.cap.release()
if self.count == self.nf: # last video if self.count == self.nf: # last video