Streaming --save-txt bug fix (#1672)
* Streaming --save-txt bug fix * cleanup
This commit is contained in:
1 parent
bc52ea2d5f
commit
54043a9fa4
2 files changed
+11
-11
No files matched your search
+2
-2
@@ -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):
|
||||
|
||||
Reference in new issue
Block a user