ultralytics 8.3.63__py3-none-any.whl → 8.3.64__py3-none-any.whl

ultralytics/__init__.py

@@ -1,6 +1,6 @@
 # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
 
-__version__ = "8.3.63"
+__version__ = "8.3.64"
 
 import os
 

ultralytics/engine/tuner.py

@@ -101,7 +101,8 @@ class Tuner:
             "copy_paste": (0.0, 1.0),  # segment copy-paste (probability)
         }
         self.args = get_cfg(overrides=args)
-        self.tune_dir = get_save_dir(self.args, name="tune")
+        self.tune_dir = get_save_dir(self.args, name=self.args.name or "tune")
+        self.args.name = None  # reset to not affect training directory
         self.tune_csv = self.tune_dir / "tune_results.csv"
         self.callbacks = _callbacks or callbacks.get_default_callbacks()
         self.prefix = colorstr("Tuner: ")

ultralytics/nn/modules/block.py

@@ -1139,10 +1139,10 @@ class TorchVision(nn.Module):
         else:
             self.m = torchvision.models.__dict__[model](pretrained=bool(weights))
         if unwrap:
-            layers = list(self.m.children())[:-truncate]
+            layers = list(self.m.children())
             if isinstance(layers[0], nn.Sequential):  # Second-level for some models like EfficientNet, Swin
                 layers = [*list(layers[0].children()), *layers[1:]]
-            self.m = nn.Sequential(*layers)
+            self.m = nn.Sequential(*(layers[:-truncate] if truncate else layers))
             self.split = split
         else:
             self.split = False

ultralytics/nn/tasks.py

@@ -955,7 +955,13 @@ def parse_model(d, ch, verbose=True):  # model_dict, input_channels(3)
     ch = [ch]
     layers, save, c2 = [], [], ch[-1]  # layers, savelist, ch out
     for i, (f, n, m, args) in enumerate(d["backbone"] + d["head"]):  # from, number, module, args
-        m = getattr(torch.nn, m[3:]) if "nn." in m else globals()[m]  # get module
+        m = (
+            getattr(torch.nn, m[3:])
+            if "nn." in m
+            else getattr(__import__("torchvision").ops, m[16:])
+            if "torchvision.ops." in m
+            else globals()[m]
+        )  # get module
         for j, a in enumerate(args):
             if isinstance(a, str):
                 with contextlib.suppress(ValueError):

ultralytics/utils/__init__.py

@@ -577,6 +577,16 @@ def is_jupyter():
     return IS_COLAB or IS_KAGGLE
 
 
+def is_runpod():
+    """
+    Check if the current script is running inside a RunPod container.
+
+    Returns:
+        (bool): True if running in RunPod, False otherwise.
+    """
+    return "RUNPOD_POD_ID" in os.environ
+
+
 def is_docker() -> bool:
     """
     Determine if the script is running inside a Docker container.
@@ -1042,7 +1052,7 @@ def set_sentry():
         auto_enabling_integrations=False,
         traces_sample_rate=1.0,
         release=__version__,
-        environment="production",  # 'dev' or 'production'
+        environment="runpod" if is_runpod() else "production",
         before_send=before_send,
         ignore_errors=[KeyboardInterrupt, FileNotFoundError],
     )

ultralytics/utils/checks.py

@@ -787,11 +787,12 @@ def is_sudo_available() -> bool:
     Check if the sudo command is available in the environment.
 
     Returns:
-        bool: True if the sudo command is available, False otherwise.
+        (bool): True if the sudo command is available, False otherwise.
     """
     if WINDOWS:
         return False
-    return subprocess.run(["sudo", "--version"], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL).returncode == 0
+    cmd = "sudo --version"
+    return subprocess.run(cmd, shell=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL).returncode == 0
 
 
 # Run checks and define constants

ultralytics/utils/tuner.py

@@ -1,6 +1,6 @@
 # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
 
-from ultralytics.cfg import TASK2DATA, TASK2METRIC, get_save_dir
+from ultralytics.cfg import TASK2DATA, TASK2METRIC, get_cfg, get_save_dir
 from ultralytics.utils import DEFAULT_CFG, DEFAULT_CFG_DICT, LOGGER, NUM_THREADS, checks
 
 
@@ -134,7 +134,9 @@ def run_ray_tune(
     tuner_callbacks = [WandbLoggerCallback(project="YOLOv8-tune")] if wandb else []
 
     # Create the Ray Tune hyperparameter search tuner
-    tune_dir = get_save_dir(DEFAULT_CFG, name="tune").resolve()  # must be absolute dir
+    tune_dir = get_save_dir(
+        get_cfg(DEFAULT_CFG, train_args), name=train_args.pop("name", "tune")
+    ).resolve()  # must be absolute dir
     tune_dir.mkdir(parents=True, exist_ok=True)
     tuner = tune.Tuner(
         trainable_with_resources,

{ultralytics-8.3.63.dist-info → ultralytics-8.3.64.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: ultralytics
-Version: 8.3.63
+Version: 8.3.64
 Summary: Ultralytics YOLO 🚀 for SOTA object detection, multi-object tracking, instance segmentation, pose estimation and image classification.
 Author-email: Glenn Jocher <glenn.jocher@ultralytics.com>, Jing Qiu <jing.qiu@ultralytics.com>
 Maintainer-email: Ultralytics <hello@ultralytics.com>

{ultralytics-8.3.63.dist-info → ultralytics-8.3.64.dist-info}/RECORD

@@ -7,7 +7,7 @@ tests/test_exports.py,sha256=bvi2GgwxiAIpnAEQQ4iImffA7zke65_CDwoXOF7anuE,8797
 tests/test_integrations.py,sha256=p3DMnnPMKsV0Qm82JVJUIY1UZ67xRgF9E8AaL76TEHE,6154
 tests/test_python.py,sha256=tW-EFJC2rjl_DvAa8khXGWYdypseQjrLjGHhe2p9r9A,23238
 tests/test_solutions.py,sha256=aY0G3vNzXGCENG9FD76MfUp7jgzeESPsUvbvQYBUvH0,4205
-ultralytics/__init__.py,sha256=sEEOEjr5AbRIanmikSj9X77osBrjhBQkbM_yZAUJjFw,709
+ultralytics/__init__.py,sha256=re8M3cQOdKVR3CsCTM1XB6E6QzFTFkBBZKw8_a9ZLao,709
 ultralytics/assets/bus.jpg,sha256=wCAZxJecGR63Od3ZRERe9Aja1Weayrb9Ug751DS_vGM,137419
 ultralytics/assets/zidane.jpg,sha256=Ftc4aeMmen1O0A3o6GCDO9FlfBslLpTAw0gnetx7bts,50427
 ultralytics/cfg/__init__.py,sha256=9xBYqfr4VxaDx3MkyK0ePIOr8iZ7BfCC2H1VNqfFInQ,39601
@@ -107,7 +107,7 @@ ultralytics/engine/model.py,sha256=tgzwnspighsesavuOEpirHS_5VPukQinj9sBAUnXpUE,5
 ultralytics/engine/predictor.py,sha256=I66hWs-SV6Yu8QTsrr9ZEWqfDblKdQkJaIqyHoFHgeQ,17740
 ultralytics/engine/results.py,sha256=sf_JPDTp9lCrcjdloTsZOBbSfzX5oVW9CL6PYf-uW8U,75114
 ultralytics/engine/trainer.py,sha256=1zT-C9YeByROTO8nEq38fzGYMt15sizkAuDHcRYEcu8,37434
-ultralytics/engine/tuner.py,sha256=IQcKuaCaJ-Toe5xA2DUvlFN577PiS2NGDU8YFbWLqO0,11886
+ultralytics/engine/tuner.py,sha256=fY-zuAGN8r2bkRVoxxLAz73DfA5CLojbiDP7TEUsAR0,11976
 ultralytics/engine/validator.py,sha256=snmE1O0wJIX610bvSOk6pV_jf2pXEQ8IdvKVkX8_agI,14901
 ultralytics/hub/__init__.py,sha256=1ifzSYV0PIT4ZWOm2V7HnpGyY3G3hCz0malw3AXHFlY,5660
 ultralytics/hub/auth.py,sha256=akS7QMg93L_cBjDGOc0Jns5-m3ao_VzBCcyKLb4f0sI,5569
@@ -173,10 +173,10 @@ ultralytics/models/yolo/world/train.py,sha256=6PVmQ0G-22OOPPwP_rqSobe2LM6e2b_lC7
 ultralytics/models/yolo/world/train_world.py,sha256=sCtg4Hnq9Y7amYjlQsdvTHXH8cKSooipvcXu_1Iyb2k,4885
 ultralytics/nn/__init__.py,sha256=rjociYD9lo_K-d-1s6TbdWklPLjTcEHk7OIlRDJstIE,615
 ultralytics/nn/autobackend.py,sha256=ONLwYlT8tu6_U0ghXfcNmG6eH9UWH-HDwnBQH2mbYWs,35540
-ultralytics/nn/tasks.py,sha256=KRmrjqWcCPgZEnb3O6Nj8Gutl-LUsWwxY_l-cIotS5o,48559
+ultralytics/nn/tasks.py,sha256=bmxRb1_MZQGWNdHcuN6QeXPRoYnqPNZURT1O9vYkMh4,48710
 ultralytics/nn/modules/__init__.py,sha256=02dPoAMtpPNQdHXHmvJeWZvJ_WG6eqwH8atLdFWgcuY,2713
 ultralytics/nn/modules/activation.py,sha256=oRkhMdqlNpIxQb35pTSUeHV-h0VyLl96GOqvIZ4OvT8,923
-ultralytics/nn/modules/block.py,sha256=HauBQiSIH9JFmT-gnk_8FNJfYfhW2heoY3iU6AdIjv0,43988
+ultralytics/nn/modules/block.py,sha256=JiT7flwBw7P8DbHf-UUHqz4uGC6WYlMrXx2f9Vlsvx0,44014
 ultralytics/nn/modules/conv.py,sha256=P3R8xTuE7GMJ2DJ_ye7D54Gh56tfh4ykNnIXBG2QcNQ,13140
 ultralytics/nn/modules/head.py,sha256=RYT31wplr64yDSHLpEZy3fyqg9W8HWlXWKrltwpqGiQ,27962
 ultralytics/nn/modules/transformer.py,sha256=fdc4xam82Dk8etahkhlc5RHW6dfY00klKj2od4QpdQo,18097
@@ -204,10 +204,10 @@ ultralytics/trackers/utils/__init__.py,sha256=lm6MckFYCPTbqIoX7w0s_daxdjNeBeKW6D
 ultralytics/trackers/utils/gmc.py,sha256=kU54RozuGJcAVlyb5_HjXiNIUIX5VuH613AMc6Gdnwg,14597
 ultralytics/trackers/utils/kalman_filter.py,sha256=OBvemZXptgn9v1sgBLvFomCqOWwjIB3-8wBbc8nakHo,21377
 ultralytics/trackers/utils/matching.py,sha256=jOGfDMaWUOGHy3e_SF-vn7m0t8nUgLNsiBwhusdkF2g,7106
-ultralytics/utils/__init__.py,sha256=2bVRKAmCrlTYigV51YI0wXM4U4sHPWXcyjTxiuhBVxw,49330
+ultralytics/utils/__init__.py,sha256=n-cHhNxlVBmRl0uWjJtxnCcU1D4esvOyJZu6mYn5obU,49554
 ultralytics/utils/autobatch.py,sha256=zc81HlAMArPASEbExty0E_zpITF8PVwin7w-xBFFZ5w,5048
 ultralytics/utils/benchmarks.py,sha256=Njv__APyz_q39wpwVBPC7qTbFBLEWNx_72arkPxo_6U,25671
-ultralytics/utils/checks.py,sha256=BbOVv3aJLhjFiS2F1Dgq3ztAvc5fcQTCqJ_B1Sb-8xc,30506
+ultralytics/utils/checks.py,sha256=nCCYwMZRHsLq-At6dVNlQrrDAFN2oj--uPjLKXMtkk4,30529
 ultralytics/utils/dist.py,sha256=fuiJQEnyyL-SighlI3hUlZPaaSreUl4Q39snF6OhQtI,2386
 ultralytics/utils/downloads.py,sha256=n5htu7rIp6izCeObZLjwaIz8t4RqoivmC2pYEa63KJo,21999
 ultralytics/utils/errors.py,sha256=sXKDEd8ws3L-yIfG_-P_h86axbm37sJNha7kFBJbQMQ,844
@@ -221,7 +221,7 @@ ultralytics/utils/plotting.py,sha256=cl8mctrkBMMTE976yrqDn1I8dH6IPO3ROZl99t5fo9w
 ultralytics/utils/tal.py,sha256=DO-c006HEI62pcrNRpmt4lpqJPC5yu3veRDOvUuExno,18498
 ultralytics/utils/torch_utils.py,sha256=JQ8HvqIVD-iL640vPBD4e9uOihhVhMuFL2G-BR-AxSM,33155
 ultralytics/utils/triton.py,sha256=2L1_rZ8xCJEjexRVj75g9YU-u4tQln_DJ5N1Yr_0bSs,4071
-ultralytics/utils/tuner.py,sha256=zhYAvVlUO4Bkijoscwlf_IKyE2FSJpFep_esuFLrG0c,6226
+ultralytics/utils/tuner.py,sha256=V_FCt0NNLhRRIgAW35oAXAkc5tbjgX1cBcFeLyJnEJs,6294
 ultralytics/utils/callbacks/__init__.py,sha256=hzL63Rce6VkZhP4Lcim9LKjadixaQG86nKqPhk7IkS0,242
 ultralytics/utils/callbacks/base.py,sha256=nbeSPjPCOb0M6FsFQ5-uFxXVzUYwmFyE4wFoA66Jpug,5803
 ultralytics/utils/callbacks/clearml.py,sha256=JH70T1OLPd9GSvC6HnaKkZHTr8fyE9RRcz3ukL62QPw,5961
@@ -233,9 +233,9 @@ ultralytics/utils/callbacks/neptune.py,sha256=waZ_bRu0-qBKujTLuqonC2gx2DkgBuVnfq
 ultralytics/utils/callbacks/raytune.py,sha256=TbuZlDb721aIkh-nMozZcP2g_ttUh2cG5LUaXmept6g,728
 ultralytics/utils/callbacks/tensorboard.py,sha256=JHOEVlNQ5dYJPd4Z-EvqbXowuK5uA0p8wPgyyaIUQs0,4194
 ultralytics/utils/callbacks/wb.py,sha256=ayhT2y62AcSOacnawshATU0rWrlSFQ77mrGgBdRl3W4,7086
-ultralytics-8.3.63.dist-info/LICENSE,sha256=DZak_2itbUtvHzD3E7GNUYSRK6jdOJ-GqncQ2weavLA,34523
-ultralytics-8.3.63.dist-info/METADATA,sha256=H2F-IYlnZkUf_fpdzSXQS76dTk0Jkld2FdRiVPqgvFo,35284
-ultralytics-8.3.63.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-ultralytics-8.3.63.dist-info/entry_points.txt,sha256=YM_wiKyTe9yRrsEfqvYolNO5ngwfoL4-NwgKzc8_7sI,93
-ultralytics-8.3.63.dist-info/top_level.txt,sha256=XP49TwiMw4QGsvTLSYiJhz1xF_k7ev5mQ8jJXaXi45Q,12
-ultralytics-8.3.63.dist-info/RECORD,,
+ultralytics-8.3.64.dist-info/LICENSE,sha256=DZak_2itbUtvHzD3E7GNUYSRK6jdOJ-GqncQ2weavLA,34523
+ultralytics-8.3.64.dist-info/METADATA,sha256=ymu6RmD6CMHaTLGep7T3NZZeLeZazQzXknqYSt50wXA,35284
+ultralytics-8.3.64.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+ultralytics-8.3.64.dist-info/entry_points.txt,sha256=YM_wiKyTe9yRrsEfqvYolNO5ngwfoL4-NwgKzc8_7sI,93
+ultralytics-8.3.64.dist-info/top_level.txt,sha256=XP49TwiMw4QGsvTLSYiJhz1xF_k7ev5mQ8jJXaXi45Q,12
+ultralytics-8.3.64.dist-info/RECORD,,