Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2019-09-10 15:34:36 +02:00
parent f20a03e28e
commit adb4894626
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
utils/datasets.py
View File

@ -147,7 +147,7 @@ class LoadWebcam: # for inference
def __next__(self): def __next__(self):
self.count += 1 self.count += 1
if cv2.waitKey(1) == 27: # esc to quit if cv2.waitKey(1) == ord('q'): # q to quit
self.cap.release() self.cap.release()
cv2.destroyAllWindows() cv2.destroyAllWindows()
raise StopIteration raise StopIteration
@ -168,7 +168,7 @@ class LoadWebcam: # for inference
# Print # Print
assert ret_val, 'Camera Error %s' % self.pipe assert ret_val, 'Camera Error %s' % self.pipe
img_path = 'webcam_%g.jpg' % self.count img_path = 'webcam.jpg'
print('webcam %g: ' % self.count, end='') print('webcam %g: ' % self.count, end='')
# Padded resize # Padded resize
@ -222,7 +222,7 @@ class LoadStreams: # multiple IP or RTSP cameras
def __next__(self): def __next__(self):
self.count += 1 self.count += 1
img0 = self.imgs.copy() img0 = self.imgs.copy()
if cv2.waitKey(1) == ord('q'): # 'q' to quit if cv2.waitKey(1) == ord('q'): # q to quit
cv2.destroyAllWindows() cv2.destroyAllWindows()
raise StopIteration raise StopIteration