diff --git a/ultralytics/__init__.py b/ultralytics/__init__.py index 060cafa6..53b6e640 100644 --- a/ultralytics/__init__.py +++ b/ultralytics/__init__.py @@ -1,6 +1,6 @@ # Ultralytics YOLO 🚀, AGPL-3.0 license -__version__ = '8.0.217' +__version__ = '8.0.218' from ultralytics.models import RTDETR, SAM, YOLO from ultralytics.models.fastsam import FastSAM diff --git a/ultralytics/data/loaders.py b/ultralytics/data/loaders.py index 78ef8500..2545e9ab 100644 --- a/ultralytics/data/loaders.py +++ b/ultralytics/data/loaders.py @@ -67,11 +67,16 @@ class LoadStreams: self.mode = 'stream' self.imgsz = imgsz self.vid_stride = vid_stride # video frame-rate stride + sources = Path(sources).read_text().rsplit() if os.path.isfile(sources) else [sources] n = len(sources) - self.sources = [ops.clean_str(x) for x in sources] # clean source names for later - self.imgs, self.fps, self.frames, self.threads, self.shape = [[]] * n, [0] * n, [0] * n, [None] * n, [[]] * n + self.fps = [0] * n # frames per second + self.frames = [0] * n + self.threads = [None] * n self.caps = [None] * n # video capture objects + self.imgs = [[] for _ in range(n)] # images + self.shape = [[] for _ in range(n)] # image shapes + self.sources = [ops.clean_str(x) for x in sources] # clean source names for later for i, s in enumerate(sources): # index, source # Start thread to read frames from video stream st = f'{i + 1}/{n}: {s}... '