Streaming --save-txt bug fix (#1672)

* Streaming --save-txt bug fix

* cleanup
This commit is contained in:
Glenn Jocher authored and GitHub committed 2020-12-11 15:45:32 -08:00
1 parent bc52ea2d5f
commit 54043a9fa4
2 files changed
+11 -11

No files matched your search

+2 -2
View File
@@ -138,7 +138,7 @@ class LoadImages: # for inference
self.files = images + videos
self.nf = ni + nv # number of files
self.video_flag = [False] * ni + [True] * nv
self.mode = 'images'
self.mode = 'image'
if any(videos):
self.new_video(videos[0]) # new video
else:
@@ -256,7 +256,7 @@ class LoadWebcam: # for inference
class LoadStreams: # multiple IP or RTSP cameras
def __init__(self, sources='streams.txt', img_size=640):
self.mode = 'images'
self.mode = 'stream'
self.img_size = img_size
if os.path.isfile(sources):