mirrors
/
yolov5
mirror of https://github.com/ultralytics/yolov5.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Update `LoadImages` `ret_val=False` handling (#5852) 

Video errors may occur.
pull/5856/head
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