ultralytics 8.0.115__py3-none-any.whl → 8.0.116__py3-none-any.whl

ultralytics/__init__.py

@@ -1,11 +1,12 @@
 # Ultralytics YOLO 🚀, AGPL-3.0 license
 
-__version__ = '8.0.115'
+__version__ = '8.0.116'
 
 from ultralytics.hub import start
 from ultralytics.vit.rtdetr import RTDETR
 from ultralytics.vit.sam import SAM
 from ultralytics.yolo.engine.model import YOLO
+from ultralytics.yolo.nas import NAS
 from ultralytics.yolo.utils.checks import check_yolo as checks
 
-__all__ = '__version__', 'YOLO', 'SAM', 'RTDETR', 'checks', 'start'  # allow simpler import
+__all__ = '__version__', 'YOLO', 'NAS', 'SAM', 'RTDETR', 'checks', 'start'  # allow simpler import

ultralytics/vit/rtdetr/model.py

@@ -110,3 +110,12 @@ class RTDETR:
         if args.batch == DEFAULT_CFG.batch:
             args.batch = 1  # default to 1 if not modified
         return Exporter(overrides=args)(model=self.model)
+
+    def __call__(self, source=None, stream=False, **kwargs):
+        """Calls the 'predict' function with given arguments to perform object detection."""
+        return self.predict(source, stream, **kwargs)
+
+    def __getattr__(self, attr):
+        """Raises error if object has no requested attribute."""
+        name = self.__class__.__name__
+        raise AttributeError(f"'{name}' object has no attribute '{attr}'. See valid attributes below.\n{self.__doc__}")

ultralytics/vit/sam/model.py

@@ -35,6 +35,15 @@ class SAM:
         """Run validation given dataset."""
         raise NotImplementedError("SAM models don't support validation")
 
+    def __call__(self, source=None, stream=False, **kwargs):
+        """Calls the 'predict' function with given arguments to perform object detection."""
+        return self.predict(source, stream, **kwargs)
+
+    def __getattr__(self, attr):
+        """Raises error if object has no requested attribute."""
+        name = self.__class__.__name__
+        raise AttributeError(f"'{name}' object has no attribute '{attr}'. See valid attributes below.\n{self.__doc__}")
+
     def info(self, detailed=False, verbose=True):
         """
         Logs model info.

ultralytics/yolo/data/utils.py

@@ -138,7 +138,7 @@ def polygon2mask(imgsz, polygons, color=1, downsample_ratio=1):
     """
     Args:
         imgsz (tuple): The image size.
-        polygons (np.ndarray): [N, M], N is the number of polygons, M is the number of points(Be divided by 2).
+        polygons (list[np.ndarray]): [N, M], N is the number of polygons, M is the number of points(Be divided by 2).
         color (int): color
         downsample_ratio (int): downsample ratio
     """

ultralytics/yolo/nas/__init__.py

@@ -0,0 +1,7 @@
+# Ultralytics YOLO 🚀, AGPL-3.0 license
+
+from .model import NAS
+from .predict import NASPredictor
+from .val import NASValidator
+
+__all__ = 'NASPredictor', 'NASValidator', 'NAS'

ultralytics/yolo/nas/model.py

@@ -0,0 +1,125 @@
+# Ultralytics YOLO 🚀, AGPL-3.0 license
+"""
+# NAS model interface
+"""
+
+from pathlib import Path
+
+import torch
+
+from ultralytics.yolo.cfg import get_cfg
+from ultralytics.yolo.engine.exporter import Exporter
+from ultralytics.yolo.utils import DEFAULT_CFG, DEFAULT_CFG_DICT, LOGGER, ROOT, is_git_dir
+from ultralytics.yolo.utils.checks import check_imgsz
+
+from ...yolo.utils.torch_utils import model_info, smart_inference_mode
+from .predict import NASPredictor
+from .val import NASValidator
+
+
+class NAS:
+
+    def __init__(self, model='yolo_nas_s.pt') -> None:
+        # Load or create new NAS model
+        import super_gradients
+
+        self.predictor = None
+        suffix = Path(model).suffix
+        if suffix == '.pt':
+            self._load(model)
+        elif suffix == '':
+            self.model = super_gradients.training.models.get(model, pretrained_weights='coco')
+        self.task = 'detect'
+        self.model.args = DEFAULT_CFG_DICT  # attach args to model
+
+        # Standardize model
+        self.model.fuse = lambda verbose: self.model
+        self.model.stride = torch.tensor([32])
+        self.model.names = dict(enumerate(self.model._class_names))
+        self.model.is_fused = lambda: False  # for info()
+        self.model.yaml = {}  # for info()
+        self.info()
+
+    @smart_inference_mode()
+    def _load(self, weights: str):
+        self.model = torch.load(weights)
+
+    @smart_inference_mode()
+    def predict(self, source=None, stream=False, **kwargs):
+        """
+        Perform prediction using the YOLO model.
+
+        Args:
+            source (str | int | PIL | np.ndarray): The source of the image to make predictions on.
+                          Accepts all source types accepted by the YOLO model.
+            stream (bool): Whether to stream the predictions or not. Defaults to False.
+            **kwargs : Additional keyword arguments passed to the predictor.
+                       Check the 'configuration' section in the documentation for all available options.
+
+        Returns:
+            (List[ultralytics.yolo.engine.results.Results]): The prediction results.
+        """
+        if source is None:
+            source = ROOT / 'assets' if is_git_dir() else 'https://ultralytics.com/images/bus.jpg'
+            LOGGER.warning(f"WARNING ⚠️ 'source' is missing. Using 'source={source}'.")
+        overrides = dict(conf=0.25, task='detect', mode='predict')
+        overrides.update(kwargs)  # prefer kwargs
+        if not self.predictor:
+            self.predictor = NASPredictor(overrides=overrides)
+            self.predictor.setup_model(model=self.model)
+        else:  # only update args if predictor is already setup
+            self.predictor.args = get_cfg(self.predictor.args, overrides)
+        return self.predictor(source, stream=stream)
+
+    def train(self, **kwargs):
+        """Function trains models but raises an error as NAS models do not support training."""
+        raise NotImplementedError("NAS models don't support training")
+
+    def val(self, **kwargs):
+        """Run validation given dataset."""
+        overrides = dict(task='detect', mode='val')
+        overrides.update(kwargs)  # prefer kwargs
+        args = get_cfg(cfg=DEFAULT_CFG, overrides=overrides)
+        args.imgsz = check_imgsz(args.imgsz, max_dim=1)
+        validator = NASValidator(args=args)
+        validator(model=self.model)
+        self.metrics = validator.metrics
+        return validator.metrics
+
+    @smart_inference_mode()
+    def export(self, **kwargs):
+        """
+        Export model.
+
+        Args:
+            **kwargs : Any other args accepted by the predictors. To see all args check 'configuration' section in docs
+        """
+        overrides = dict(task='detect')
+        overrides.update(kwargs)
+        overrides['mode'] = 'export'
+        args = get_cfg(cfg=DEFAULT_CFG, overrides=overrides)
+        args.task = self.task
+        if args.imgsz == DEFAULT_CFG.imgsz:
+            args.imgsz = self.model.args['imgsz']  # use trained imgsz unless custom value is passed
+        if args.batch == DEFAULT_CFG.batch:
+            args.batch = 1  # default to 1 if not modified
+        return Exporter(overrides=args)(model=self.model)
+
+    def info(self, detailed=False, verbose=True):
+        """
+        Logs model info.
+
+        Args:
+            detailed (bool): Show detailed information about model.
+            verbose (bool): Controls verbosity.
+        """
+        return model_info(self.model, detailed=detailed, verbose=verbose, imgsz=640)
+
+    def __call__(self, source=None, stream=False, **kwargs):
+        """Calls the 'predict' function with given arguments to perform object detection."""
+        return self.predict(source, stream, **kwargs)
+
+    def __getattr__(self, attr):
+        """Raises error if object has no requested attribute."""
+        name = self.__class__.__name__
+        raise AttributeError(f"'{name}' object has no attribute '{attr}'. See valid attributes below.\n{self.__doc__}")

ultralytics/yolo/nas/predict.py

@@ -0,0 +1,35 @@
+# Ultralytics YOLO 🚀, AGPL-3.0 license
+
+import torch
+
+from ultralytics.yolo.engine.predictor import BasePredictor
+from ultralytics.yolo.engine.results import Results
+from ultralytics.yolo.utils import ops
+from ultralytics.yolo.utils.ops import xyxy2xywh
+
+
+class NASPredictor(BasePredictor):
+
+    def postprocess(self, preds_in, img, orig_imgs):
+        """Postprocesses predictions and returns a list of Results objects."""
+
+        # Cat boxes and class scores
+        boxes = xyxy2xywh(preds_in[0][0])
+        preds = torch.cat((boxes, preds_in[0][1]), -1).permute(0, 2, 1)
+
+        preds = ops.non_max_suppression(preds,
+                                        self.args.conf,
+                                        self.args.iou,
+                                        agnostic=self.args.agnostic_nms,
+                                        max_det=self.args.max_det,
+                                        classes=self.args.classes)
+
+        results = []
+        for i, pred in enumerate(preds):
+            orig_img = orig_imgs[i] if isinstance(orig_imgs, list) else orig_imgs
+            if not isinstance(orig_imgs, torch.Tensor):
+                pred[:, :4] = ops.scale_boxes(img.shape[2:], pred[:, :4], orig_img.shape)
+            path = self.batch[0]
+            img_path = path[i] if isinstance(path, list) else path
+            results.append(Results(orig_img=orig_img, path=img_path, names=self.model.names, boxes=pred))
+        return results

ultralytics/yolo/nas/val.py

@@ -0,0 +1,25 @@
+# Ultralytics YOLO 🚀, AGPL-3.0 license
+
+import torch
+
+from ultralytics.yolo.utils import ops
+from ultralytics.yolo.utils.ops import xyxy2xywh
+from ultralytics.yolo.v8.detect import DetectionValidator
+
+__all__ = ['NASValidator']
+
+
+class NASValidator(DetectionValidator):
+
+    def postprocess(self, preds_in):
+        """Apply Non-maximum suppression to prediction outputs."""
+        boxes = xyxy2xywh(preds_in[0][0])
+        preds = torch.cat((boxes, preds_in[0][1]), -1).permute(0, 2, 1)
+        return ops.non_max_suppression(preds,
+                                       self.args.conf,
+                                       self.args.iou,
+                                       labels=self.lb,
+                                       multi_label=False,
+                                       agnostic=self.args.single_cls,
+                                       max_det=self.args.max_det,
+                                       max_time_img=0.5)

ultralytics/yolo/utils/__init__.py

@@ -224,6 +224,11 @@ def set_logging(name=LOGGING_NAME, verbose=True):
                 'propagate': False}}})
 
 
+def emojis(string=''):
+    """Return platform-dependent emoji-safe version of string."""
+    return string.encode().decode('ascii', 'ignore') if WINDOWS else string
+
+
 class EmojiFilter(logging.Filter):
     """
     A custom logging filter class for removing emojis in log messages.
@@ -533,6 +538,7 @@ def get_user_config_dir(sub_dir='Ultralytics'):
     # GCP and AWS lambda fix, only /tmp is writeable
     if not is_dir_writeable(str(path.parent)):
         path = Path('/tmp') / sub_dir
+        LOGGER.warning(f"WARNING ⚠️ user config directory is not writeable, defaulting to '{path}'.")
 
     # Create the subdirectory if it does not exist
     path.mkdir(parents=True, exist_ok=True)
@@ -544,11 +550,6 @@ USER_CONFIG_DIR = Path(os.getenv('YOLO_CONFIG_DIR', get_user_config_dir()))  # U
 SETTINGS_YAML = USER_CONFIG_DIR / 'settings.yaml'
 
 
-def emojis(string=''):
-    """Return platform-dependent emoji-safe version of string."""
-    return string.encode().decode('ascii', 'ignore') if WINDOWS else string
-
-
 def colorstr(*input):
     """Colors a string https://en.wikipedia.org/wiki/ANSI_escape_code, i.e.  colorstr('blue', 'hello world')."""
     *args, string = input if len(input) > 1 else ('blue', 'bold', input[0])  # color arguments, string

ultralytics/yolo/utils/benchmarks.py

@@ -90,7 +90,7 @@ def benchmark(model=Path(SETTINGS['weights_dir']) / 'yolov8n.pt',
                 filename = model.ckpt_path or model.cfg
                 export = model  # PyTorch format
             else:
-                filename = model.export(imgsz=imgsz, format=format, half=half, int8=int8, device=device)  # all others
+                filename = model.export(imgsz=imgsz, format=format, half=half, int8=int8, device=device, verbose=False)
                 export = YOLO(filename, task=model.task)
                 assert suffix in str(filename), 'export failed'
             emoji = '❎'  # indicates export succeeded
@@ -196,8 +196,17 @@ class ProfileModels:
                 model.fuse()  # to report correct params and GFLOPs in model.info()
                 model_info = model.info()
                 if self.trt and self.device.type != 'cpu' and not engine_file.is_file():
-                    engine_file = model.export(format='engine', half=True, imgsz=self.imgsz, device=self.device)
-                onnx_file = model.export(format='onnx', half=True, imgsz=self.imgsz, simplify=True, device=self.device)
+                    engine_file = model.export(format='engine',
+                                               half=True,
+                                               imgsz=self.imgsz,
+                                               device=self.device,
+                                               verbose=False)
+                onnx_file = model.export(format='onnx',
+                                         half=True,
+                                         imgsz=self.imgsz,
+                                         simplify=True,
+                                         device=self.device,
+                                         verbose=False)
             elif file.suffix == '.onnx':
                 model_info = self.get_onnx_model_info(file)
                 onnx_file = file
@@ -254,7 +263,7 @@ class ProfileModels:
         for _ in range(3):
             start_time = time.time()
             for _ in range(self.num_warmup_runs):
-                model(input_data, verbose=False)
+                model(input_data, imgsz=self.imgsz, verbose=False)
             elapsed = time.time() - start_time
 
         # Compute number of runs as higher of min_time or num_timed_runs
@@ -263,7 +272,7 @@ class ProfileModels:
         # Timed runs
         run_times = []
         for _ in tqdm(range(num_runs), desc=engine_file):
-            results = model(input_data, verbose=False)
+            results = model(input_data, imgsz=self.imgsz, verbose=False)
             run_times.append(results[0].speed['inference'])  # Convert to milliseconds
 
         run_times = self.iterative_sigma_clipping(np.array(run_times), sigma=2, max_iters=3)  # sigma clipping

ultralytics/yolo/utils/callbacks/dvc.py

@@ -4,7 +4,7 @@ import os
 import pkg_resources as pkg
 
 from ultralytics.yolo.utils import LOGGER, TESTS_RUNNING
-from ultralytics.yolo.utils.torch_utils import get_flops, get_num_params
+from ultralytics.yolo.utils.torch_utils import model_info_for_loggers
 
 try:
     from importlib.metadata import version
@@ -42,7 +42,7 @@ def _log_images(image_path, prefix=''):
 def _log_plots(plots, prefix=''):
     for name, params in plots.items():
         timestamp = params['timestamp']
-        if _processed_plots.get(name, None) != timestamp:
+        if _processed_plots.get(name) != timestamp:
             _log_images(name, prefix)
             _processed_plots[name] = timestamp
 
@@ -100,12 +100,7 @@ def on_fit_epoch_end(trainer):
             live.log_metric(metric, value)
 
         if trainer.epoch == 0:
-            model_info = {
-                'model/parameters': get_num_params(trainer.model),
-                'model/GFLOPs': round(get_flops(trainer.model), 3),
-                'model/speed(ms)': round(trainer.validator.speed['inference'], 3)}
-
-            for metric, value in model_info.items():
+            for metric, value in model_info_for_loggers(trainer).items():
                 live.log_metric(metric, value, plot=False)
 
         _log_plots(trainer.plots, 'train')

ultralytics/yolo/utils/checks.py

@@ -19,9 +19,9 @@ import requests
 import torch
 from matplotlib import font_manager
 
-from ultralytics.yolo.utils import (AUTOINSTALL, LOGGER, ONLINE, ROOT, USER_CONFIG_DIR, TryExcept, clean_url, colorstr,
-                                    downloads, emojis, is_colab, is_docker, is_kaggle, is_online, is_pip_package,
-                                    url2file)
+from ultralytics.yolo.utils import (AUTOINSTALL, LOGGER, ONLINE, RANK, ROOT, USER_CONFIG_DIR, TryExcept, clean_url,
+                                    colorstr, downloads, emojis, is_colab, is_docker, is_kaggle, is_online,
+                                    is_pip_package, url2file)
 
 
 def is_ascii(s) -> bool:
@@ -164,23 +164,26 @@ def check_font(font='Arial.ttf'):
     Returns:
         file (Path): Resolved font file path.
     """
-    name = Path(font).name
-
-    # Check USER_CONFIG_DIR
-    file = USER_CONFIG_DIR / name
-    if file.exists():
-        return file
-
-    # Check system fonts
-    matches = [s for s in font_manager.findSystemFonts() if font in s]
-    if any(matches):
-        return matches[0]
-
-    # Download to USER_CONFIG_DIR if missing
-    url = f'https://ultralytics.com/assets/{name}'
-    if downloads.is_url(url):
-        downloads.safe_download(url=url, file=file)
-        return file
+    from ultralytics.yolo.utils.torch_utils import torch_distributed_zero_first
+
+    with torch_distributed_zero_first(RANK):
+        name = Path(font).name
+
+        # Check USER_CONFIG_DIR
+        file = USER_CONFIG_DIR / name
+        if file.exists():
+            return file
+
+        # Check system fonts
+        matches = [s for s in font_manager.findSystemFonts() if font in s]
+        if any(matches):
+            return matches[0]
+
+        # Download to USER_CONFIG_DIR if missing
+        url = f'https://ultralytics.com/assets/{name}'
+        if downloads.is_url(url):
+            downloads.safe_download(url=url, file=file)
+            return file
 
 
 def check_python(minimum: str = '3.7.0') -> bool:

ultralytics/yolo/utils/torch_utils.py

@@ -64,6 +64,8 @@ def select_device(device='', batch=0, newline=False, verbose=True):
     if cpu or mps:
         os.environ['CUDA_VISIBLE_DEVICES'] = '-1'  # force torch.cuda.is_available() = False
     elif device:  # non-cpu device requested
+        if device == 'cuda':
+            device = '0'
         visible = os.environ.get('CUDA_VISIBLE_DEVICES', None)
         os.environ['CUDA_VISIBLE_DEVICES'] = device  # set environment variable - must be before assert is_available()
         if not (torch.cuda.is_available() and torch.cuda.device_count() >= len(device.replace(',', ''))):

{ultralytics-8.0.115.dist-info → ultralytics-8.0.116.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ultralytics
-Version: 8.0.115
+Version: 8.0.116
 Summary: Ultralytics YOLOv8 for SOTA object detection, multi-object tracking, instance segmentation, pose estimation and image classification.
 Home-page: https://github.com/ultralytics/ultralytics
 Author: Ultralytics
@@ -161,9 +161,11 @@ path = model.export(format="onnx")  # export the model to ONNX format
 
 ## <div align="center">Models</div>
 
-All YOLOv8 pretrained models are available here. Detect, Segment and Pose models are pretrained on the [COCO](https://github.com/ultralytics/ultralytics/blob/main/ultralytics/datasets/coco.yaml) dataset, while Classify models are pretrained on the [ImageNet](https://github.com/ultralytics/ultralytics/blob/main/ultralytics/datasets/ImageNet.yaml) dataset.
+YOLOv8 [Detect](https://docs.ultralytics.com/tasks/detect), [Segment](https://docs.ultralytics.com/tasks/segment) and [Pose](https://docs.ultralytics.com/tasks/pose) models pretrained on the [COCO](https://docs.ultralytics.com/datasets/detect/coco) dataset are available here, as well as YOLOv8 [Classify](https://docs.ultralytics.com/modes/classify) models pretrained on the [ImageNet](https://docs.ultralytics.com/datasets/classify/imagenet) dataset. [Track](https://docs.ultralytics.com/modes/track) mode is available for all Detect, Segment and Pose models.
 
-[Models](https://github.com/ultralytics/ultralytics/tree/main/ultralytics/models) download automatically from the latest Ultralytics [release](https://github.com/ultralytics/assets/releases) on first use.
+<img width="1024" src="https://raw.githubusercontent.com/ultralytics/assets/tasks/im/banner-tasks.png">
+
+All [Models](https://github.com/ultralytics/ultralytics/tree/main/ultralytics/models) download automatically from the latest Ultralytics [release](https://github.com/ultralytics/assets/releases) on first use.
 
 <details open><summary>Detection</summary>
 

{ultralytics-8.0.115.dist-info → ultralytics-8.0.116.dist-info}/RECORD

@@ -1,4 +1,4 @@
-ultralytics/__init__.py,sha256=Vy_A6gY4p70mT4qiooj19cMhsLen0lrqlBVnK4j4ZUU,383
+ultralytics/__init__.py,sha256=kGXynKiWGuj7CVtgwKm1YG31e0E0Ppqx104S-9rSBgk,427
 ultralytics/assets/bus.jpg,sha256=wCAZxJecGR63Od3ZRERe9Aja1Weayrb9Ug751DS_vGM,137419
 ultralytics/assets/zidane.jpg,sha256=Ftc4aeMmen1O0A3o6GCDO9FlfBslLpTAw0gnetx7bts,50427
 ultralytics/datasets/Argoverse.yaml,sha256=Q6hKRtI52JOYt4qmjkeo192mmgSkuCdOnfiUTxtBy5A,2751
@@ -59,14 +59,14 @@ ultralytics/tracker/utils/kalman_filter.py,sha256=gMc4gDKo_dkMiL-Zkt-U7mTxjK_z4A
 ultralytics/tracker/utils/matching.py,sha256=O_wJb-6ZuucIl2TuLsVe7QmqqxUNP8aoCjaWxnhk_Xg,8699
 ultralytics/vit/__init__.py,sha256=_lrglMOLuopqBVEAwGFnM8Huc6SJAHcysuGc0dmA8l0,149
 ultralytics/vit/rtdetr/__init__.py,sha256=1Zpc6ZcizFO0EMhP8X4m3DG27vDBX4aM4RX0rMSeo6E,197
-ultralytics/vit/rtdetr/model.py,sha256=1LdCA9IP6GKMA-U3L1amlwMvIQYw8vy042dbO9M8jk4,4610
+ultralytics/vit/rtdetr/model.py,sha256=CVtRbvCTQLFmG5d2CCFfVdnWrv9ClL8vaw4_BPtDrBs,5077
 ultralytics/vit/rtdetr/predict.py,sha256=zYV0c5lWr79h_J8kBhcFPB5wcDOyGdgR1qr_Z_l6Rj4,1902
 ultralytics/vit/rtdetr/val.py,sha256=6ERi9xIFOEkoEvWuy9DuOc5u1OaI2jm2Jag9l9zesvw,4790
 ultralytics/vit/sam/__init__.py,sha256=qAi91Krwqwsu2jruHiJgLabEkWo9yeZ5gDIHIKzX-PE,130
 ultralytics/vit/sam/amg.py,sha256=tzpbVPiyIwENW0xgtdy08kVGgn69zOk4VGjHmEFTCeE,13304
 ultralytics/vit/sam/autosize.py,sha256=3lym72tdiJFvbyIOMzxUzumqx82PY3lYEB3ctWNgM4s,3878
 ultralytics/vit/sam/build.py,sha256=Cp0LhlNcjLl-2OA_06eaeYlA3lK3N0KqUfhmZJ7A8Hw,3781
-ultralytics/vit/sam/model.py,sha256=ynJtuLhvhva4-69o2TDtTk7viIuOpJpVGr-u-Rf5D6E,1807
+ultralytics/vit/sam/model.py,sha256=kS-zpGw8wXKkU0yWCH-uGvmMFWB_AnEAa6NMVdsUDEA,2274
 ultralytics/vit/sam/predict.py,sha256=RcC3uLs3RiqWeW7UxdyjK6LcmqysFqRRfz3f-pV_mPQ,2139
 ultralytics/vit/sam/modules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ultralytics/vit/sam/modules/decoders.py,sha256=MJhFLPA7broz2oiYI5tUi3ROWgPxNshhKFXKUP4CEPc,6351
@@ -86,7 +86,7 @@ ultralytics/yolo/data/build.py,sha256=9eVmHXp79QpTLVB-sbaZSM6Qaet9YbiiiTW4TviAGE
 ultralytics/yolo/data/converter.py,sha256=D-O9eOcO1qg7ultRv2xLAQR39JjVIUkR--g05zquQDA,9190
 ultralytics/yolo/data/dataset.py,sha256=jmbIpgMaWhQtLfWC4hP8LBaBG2wpFGpxXDg0478FF58,13372
 ultralytics/yolo/data/dataset_wrappers.py,sha256=a5uwWLhWDRCG5VFto6I3zknxaElxnrcOHiuTAKmmmYg,1776
-ultralytics/yolo/data/utils.py,sha256=aBAp1jBzLOPwxGrelDqCi0J2kl3T1CuzpzwRJlzAviU,24050
+ultralytics/yolo/data/utils.py,sha256=nqo_SJNnlBANLEOzvmJGFxz_Ai0OvAQe-CEmMCgDUBo,24056
 ultralytics/yolo/data/dataloaders/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ultralytics/yolo/data/dataloaders/stream_loaders.py,sha256=0bg7lPloNLml7zO9Qe-fz4Cw7VbbSZjrEUbBe-lfpzc,14905
 ultralytics/yolo/data/dataloaders/v5augmentations.py,sha256=t39RSuHCDCX4MV9EkIoJWaPOjkYlNbbpi-30fg725rw,17646
@@ -98,10 +98,14 @@ ultralytics/yolo/engine/predictor.py,sha256=j1F6f6AHs-LNBta4IG45VyLXEs78UaYpjoIE
 ultralytics/yolo/engine/results.py,sha256=KNiXW_xkSN8qJJl7gqQKCz_6zjU0nyBHKqD3yoIf-2E,24287
 ultralytics/yolo/engine/trainer.py,sha256=Xie6mlF-jCESk-3tMLMf1nY8bI8_-9Owgg2EAQPfMLI,31199
 ultralytics/yolo/engine/validator.py,sha256=ZbOde8CClZQgTFXJ68v4DP_etR1prJYrwMayyKUcfTk,11715
-ultralytics/yolo/utils/__init__.py,sha256=mAUsKyvtOKkR0hcup-2IsMt9LhSa5w21c9smTSxnkmI,27849
+ultralytics/yolo/nas/__init__.py,sha256=O7qvgqJqoLB1NXwjTNHMJHJRhDwNHS2P_oyUV_b1qq8,179
+ultralytics/yolo/nas/model.py,sha256=LU0NaoZjcJEDvKQetavJ7YmVzn-kGO-IJDDw8vKz-S4,4977
+ultralytics/yolo/nas/predict.py,sha256=WbO9fxO2o_WA3nM7KkgHDgr2eJWdYsT725BcJ6_Qtrs,1465
+ultralytics/yolo/nas/val.py,sha256=D9jIFcafuD768X5l9Q56hrXIz-Jl45uEReIsPyrlNgs,953
+ultralytics/yolo/utils/__init__.py,sha256=OJ0x67npaXsLYD5TzKmlMZ-n74Jq0hG43fHglT8SMkE,27955
 ultralytics/yolo/utils/autobatch.py,sha256=-8eEMRN9KyX9nslL5XAC2gMTFAH5YFA33q6nnQDj1qU,3846
-ultralytics/yolo/utils/benchmarks.py,sha256=M2bC3dVTDyBpWMVz4FMTAI9iREp2bvTZBHc7Jre3-3U,15428
-ultralytics/yolo/utils/checks.py,sha256=dgD4pO_epK8eUN9LQB_n5cPYDY2azxh3Gj2cWzF10GI,17380
+ultralytics/yolo/utils/benchmarks.py,sha256=lFdnVqrXlI2n8JC-B1UgW2zip-Tsii7HjrJrCQmDV-g,15888
+ultralytics/yolo/utils/checks.py,sha256=3kLYaCCuZpGNKaX9RYDEHfiPgKD-Sc3Ep5kmR-LBQFI,17568
 ultralytics/yolo/utils/dist.py,sha256=qavAVcOeyLV0z31Spo1G9batXljnYzp1NN8YWFoBTxk,2596
 ultralytics/yolo/utils/downloads.py,sha256=w_Pnw_0_s6rNno54J1-qIm8Zy_I0jr7B9fSdD-TwX8g,12114
 ultralytics/yolo/utils/errors.py,sha256=u8NUGZbWVrr4O3ez6UXnabNb_zRwJUPJqDOUSpmX3-k,317
@@ -113,13 +117,13 @@ ultralytics/yolo/utils/ops.py,sha256=ne7GC7hOuAS3Lnk7eDx5w8LOMtbb_l7nJcb0wpBVJRQ
 ultralytics/yolo/utils/patches.py,sha256=lxG_qZlTxiCcKWrVFtvSVA8WULRa483fOJ4ffYyGQ50,1241
 ultralytics/yolo/utils/plotting.py,sha256=ScX-_AasukwuQpOT8q18YMSKMwJBDKFM8q1o8oY8MqE,24431
 ultralytics/yolo/utils/tal.py,sha256=BFovrLm0m5_Pv-hFzYMAD3oonS1MvDqdy51f_Cx6Aq8,13645
-ultralytics/yolo/utils/torch_utils.py,sha256=S2emgg-VTZZ0pV2YE6VMxaUgaVpA7HVdO_U2RLtvJos,22177
+ultralytics/yolo/utils/torch_utils.py,sha256=mpAdku7PNYFLOFcijSWA3poZ6ShIShOlEHlqvBDzJp4,22231
 ultralytics/yolo/utils/tuner.py,sha256=yq3nYsnug2LC-iJJwLlXrQwJJT2VanujFFKnY-06CBQ,2300
 ultralytics/yolo/utils/callbacks/__init__.py,sha256=nhrnMPpPDb5fgqw42w8e7fC5TjEPC-jp04dpQtaQtkU,214
 ultralytics/yolo/utils/callbacks/base.py,sha256=VpiMIW9qiyncMq9cLRmm5WGr38On0LVTK2XNDmliEbE,5593
 ultralytics/yolo/utils/callbacks/clearml.py,sha256=VYtuNTlB8M5rIFQupm53uhDEXe8qZrrOjXXHTI0bIUw,5902
 ultralytics/yolo/utils/callbacks/comet.py,sha256=_Dm-Qbeyi36nSSETsoiObG52gceLw6RE2TodEYW7sUY,12978
-ultralytics/yolo/utils/callbacks/dvc.py,sha256=lhZmpgaCvdtM8S_a7fjcH13EPpBt9_H-BeIR4WRxe08,4535
+ultralytics/yolo/utils/callbacks/dvc.py,sha256=7xNhDoTQqapOULEVrwxluZGOdKZVK74EPLFu4Rm47Tw,4301
 ultralytics/yolo/utils/callbacks/hub.py,sha256=Z-F48IcG2PtlJo3mnzp31YdXrS8nyMn6004-Ke6NLyM,3310
 ultralytics/yolo/utils/callbacks/mlflow.py,sha256=fCp4zNefamIzqZpOkoG2UqKVkeuIX41_em2Uw33iasY,2493
 ultralytics/yolo/utils/callbacks/neptune.py,sha256=lBvXuXkgHfiDkdeRAM7TbK7xQBSAKxYQQAwHt7SMNi8,3679
@@ -143,9 +147,9 @@ ultralytics/yolo/v8/segment/__init__.py,sha256=TOdf3ju-D5hSi-PYMpETFmv-wyhIRKGuj
 ultralytics/yolo/v8/segment/predict.py,sha256=QSwXjC7sq7l7kT0W1UrwGZrUXNqkveRGRIsRnlQlWYM,2839
 ultralytics/yolo/v8/segment/train.py,sha256=Ag0uHlJp297dljtECqZ5VVfnWj7m86oD1Dy1_s4vIEs,2499
 ultralytics/yolo/v8/segment/val.py,sha256=R3D_itui1lU0iXC1aqmZvmwGX77oJt--bQ7MhTErsBY,12906
-ultralytics-8.0.115.dist-info/LICENSE,sha256=DZak_2itbUtvHzD3E7GNUYSRK6jdOJ-GqncQ2weavLA,34523
-ultralytics-8.0.115.dist-info/METADATA,sha256=mqLuSairlVeYqtypDycYyoTaFcAbrAkBvWkKr1pYq2k,26843
-ultralytics-8.0.115.dist-info/WHEEL,sha256=pkctZYzUS4AYVn6dJ-7367OJZivF2e8RA9b_ZBjif18,92
-ultralytics-8.0.115.dist-info/entry_points.txt,sha256=Ck1F6qKNokeHozQD5pmaFgXHL6dKyC2qCdyXao2e6Yg,103
-ultralytics-8.0.115.dist-info/top_level.txt,sha256=XP49TwiMw4QGsvTLSYiJhz1xF_k7ev5mQ8jJXaXi45Q,12
-ultralytics-8.0.115.dist-info/RECORD,,
+ultralytics-8.0.116.dist-info/LICENSE,sha256=DZak_2itbUtvHzD3E7GNUYSRK6jdOJ-GqncQ2weavLA,34523
+ultralytics-8.0.116.dist-info/METADATA,sha256=fhzEqGcUjDMa-o-bqGrtVJHMhLse8F6-KqsHmd5r9Ug,27157
+ultralytics-8.0.116.dist-info/WHEEL,sha256=pkctZYzUS4AYVn6dJ-7367OJZivF2e8RA9b_ZBjif18,92
+ultralytics-8.0.116.dist-info/entry_points.txt,sha256=Ck1F6qKNokeHozQD5pmaFgXHL6dKyC2qCdyXao2e6Yg,103
+ultralytics-8.0.116.dist-info/top_level.txt,sha256=XP49TwiMw4QGsvTLSYiJhz1xF_k7ev5mQ8jJXaXi45Q,12
+ultralytics-8.0.116.dist-info/RECORD,,