ultralytics 8.1.31__py3-none-any.whl → 8.1.32__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of ultralytics might be problematic. Click here for more details.

ultralytics/__init__.py CHANGED
@@ -1,6 +1,6 @@
1
1
  # Ultralytics YOLO 🚀, AGPL-3.0 license
2
2
 
3
- __version__ = "8.1.31"
3
+ __version__ = "8.1.32"
4
4
 
5
5
  from ultralytics.data.explorer.explorer import Explorer
6
6
  from ultralytics.models import RTDETR, SAM, YOLO, YOLOWorld
@@ -243,7 +243,7 @@ def check_cfg(cfg, hard=True):
243
243
  f"'{k}={v}' is of invalid type {type(v).__name__}. "
244
244
  f"Valid '{k}' types are int (i.e. '{k}=0') or float (i.e. '{k}=0.5')"
245
245
  )
246
- cfg[k] = float(v)
246
+ cfg[k] = v = float(v)
247
247
  if not (0.0 <= v <= 1.0):
248
248
  raise ValueError(f"'{k}={v}' is an invalid value. " f"Valid '{k}' values are between 0.0 and 1.0.")
249
249
  elif k in CFG_INT_KEYS and not isinstance(v, int):
@@ -0,0 +1,24 @@
1
+ # Ultralytics YOLO 🚀, AGPL-3.0 license
2
+ # African-wildlife dataset by Ultralytics
3
+ # Documentation: https://docs.ultralytics.com/datasets/detect/african-wildlife/
4
+ # Example usage: yolo train data=african-wildlife.yaml
5
+ # parent
6
+ # ├── ultralytics
7
+ # └── datasets
8
+ # └── african-wildlife ← downloads here (100 MB)
9
+
10
+ # Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
11
+ path: ../datasets/african-wildlife # dataset root dir
12
+ train: train/images # train images (relative to 'path') 1052 images
13
+ val: valid/images # val images (relative to 'path') 225 images
14
+ test: test/images # test images (relative to 'path') 227 images
15
+
16
+ # Classes
17
+ names:
18
+ 0: buffalo
19
+ 1: elephant
20
+ 2: rhino
21
+ 3: zebra
22
+
23
+ # Download script/URL (optional)
24
+ download: https://ultralytics.com/assets/african-wildlife.zip
ultralytics/nn/tasks.py CHANGED
@@ -572,7 +572,7 @@ class WorldModel(DetectionModel):
572
572
  check_requirements("git+https://github.com/openai/CLIP.git")
573
573
  import clip
574
574
 
575
- if not self.clip_model:
575
+ if not getattr(self, "clip_model", None): # for backwards compatibility of models lacking clip_model attribute
576
576
  self.clip_model = clip.load("ViT-B/32")[0]
577
577
  device = next(self.clip_model.parameters()).device
578
578
  text_token = clip.tokenize(text).to(device)
@@ -3,7 +3,7 @@
3
3
  import subprocess
4
4
 
5
5
  from ultralytics.cfg import TASK2DATA, TASK2METRIC, get_save_dir
6
- from ultralytics.utils import DEFAULT_CFG, DEFAULT_CFG_DICT, LOGGER, NUM_THREADS
6
+ from ultralytics.utils import DEFAULT_CFG, DEFAULT_CFG_DICT, LOGGER, NUM_THREADS, checks
7
7
 
8
8
 
9
9
  def run_ray_tune(
@@ -40,7 +40,7 @@ def run_ray_tune(
40
40
  train_args = {}
41
41
 
42
42
  try:
43
- subprocess.run("pip install ray[tune]".split(), check=True)
43
+ subprocess.run("pip install ray[tune]<=2.9.3".split(), check=True) # do not add single quotes here
44
44
 
45
45
  import ray
46
46
  from ray import tune
@@ -48,7 +48,7 @@ def run_ray_tune(
48
48
  from ray.air.integrations.wandb import WandbLoggerCallback
49
49
  from ray.tune.schedulers import ASHAScheduler
50
50
  except ImportError:
51
- raise ModuleNotFoundError('Tuning hyperparameters requires Ray Tune. Install with: pip install "ray[tune]"')
51
+ raise ModuleNotFoundError('Ray Tune required but not found. To install run: pip install "ray[tune]<=2.9.3"')
52
52
 
53
53
  try:
54
54
  import wandb
@@ -57,6 +57,7 @@ def run_ray_tune(
57
57
  except (ImportError, AssertionError):
58
58
  wandb = False
59
59
 
60
+ checks.check_version(ray.__version__, "<=2.9.3", "ray")
60
61
  default_space = {
61
62
  # 'optimizer': tune.choice(['SGD', 'Adam', 'AdamW', 'NAdam', 'RAdam', 'RMSProp']),
62
63
  "lr0": tune.uniform(1e-5, 1e-1),
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: ultralytics
3
- Version: 8.1.31
3
+ Version: 8.1.32
4
4
  Summary: Ultralytics YOLOv8 for SOTA object detection, multi-object tracking, instance segmentation, pose estimation and image classification.
5
5
  Author: Glenn Jocher, Ayush Chaurasia, Jing Qiu
6
6
  Maintainer: Glenn Jocher, Ayush Chaurasia, Jing Qiu
@@ -1,7 +1,7 @@
1
- ultralytics/__init__.py,sha256=rDgevqxLqhO8NJR6otxXOqpbVQbtuhyXvkKoXK1rhL8,625
1
+ ultralytics/__init__.py,sha256=olAbfZvhpDNTg98r-aFv6S42b7g-APSxekGTY6Cmayc,625
2
2
  ultralytics/assets/bus.jpg,sha256=wCAZxJecGR63Od3ZRERe9Aja1Weayrb9Ug751DS_vGM,137419
3
3
  ultralytics/assets/zidane.jpg,sha256=Ftc4aeMmen1O0A3o6GCDO9FlfBslLpTAw0gnetx7bts,50427
4
- ultralytics/cfg/__init__.py,sha256=rNlOdOpORYyqCj97ePycRxGr-gVvmmr2KOVFdXE262Y,21258
4
+ ultralytics/cfg/__init__.py,sha256=DGqNA0FT2zZnkI09S0ttagkKopP97dQ3TVHo2nA8yFQ,21262
5
5
  ultralytics/cfg/default.yaml,sha256=jc6iBzaQIg_uohd5hHPBtYp6gJIRtYqsChwYVTRjIkI,8091
6
6
  ultralytics/cfg/datasets/Argoverse.yaml,sha256=FyeuJT5CHq_9d4hlfAf0kpZlnbUMO0S--UJ1yIqcdKk,3134
7
7
  ultralytics/cfg/datasets/DOTAv1.5.yaml,sha256=YDsyFPI6F6-OQXLBM3hOXo3vADYREwZzmMQfJNdpWyM,1193
@@ -12,6 +12,7 @@ ultralytics/cfg/datasets/Objects365.yaml,sha256=kiiV4KLMH2mcPPRrg6cQGygnbiTrHxwt
12
12
  ultralytics/cfg/datasets/SKU-110K.yaml,sha256=geRkccBRl2eKgfNYTOPYwD9mTfqktTBGiMJoE3PZEnA,2493
13
13
  ultralytics/cfg/datasets/VOC.yaml,sha256=3-CDpjIq_s5pkbsJ9TjrYIeV24rYGuJGu4Qg6uktEZE,3655
14
14
  ultralytics/cfg/datasets/VisDrone.yaml,sha256=NfrbjVnE48E7TPbxtF7rtQHvVBO0DchFJFEuGrG1VRU,3073
15
+ ultralytics/cfg/datasets/african-wildlife.yaml,sha256=vtFAaFD2wZQmt3ze-LMUOeUa-7QG57LAVwuazCbROcU,869
15
16
  ultralytics/cfg/datasets/brain-tumor.yaml,sha256=o1rX1_iw97HjiG904l4u42x13jyHOGvfmWEzz0BpOZg,795
16
17
  ultralytics/cfg/datasets/carparts-seg.yaml,sha256=pvTi3EH2j6UuG0LHoQJ7JjQv_cJoO8UKSXPptUTnl8U,1207
17
18
  ultralytics/cfg/datasets/coco-pose.yaml,sha256=w7H-J2e87GIV_PZdRDgqEFa75ObScpBK_l85U4ZMsMo,1603
@@ -139,7 +140,7 @@ ultralytics/models/yolo/segment/train.py,sha256=aOQpDIptZfKSl9mFa6B-3W3QccMRlmBI
139
140
  ultralytics/models/yolo/segment/val.py,sha256=njiF6RWddS-HOWxVvlk5PXRw6UOgEt_HEOZVPF7rruQ,11745
140
141
  ultralytics/nn/__init__.py,sha256=4BPLHY89xEM_al5uK0aOmFgiML6CMGEZbezxOvTjOEs,587
141
142
  ultralytics/nn/autobackend.py,sha256=ldDepfx4mfQVgk4w6lAqhLtxng3kchJRNSb35qMrpUg,28836
142
- ultralytics/nn/tasks.py,sha256=G0OrHmSN4QW2SlkmGSfjXzXmOWZUiCQ4QKPHb0_9SwM,42806
143
+ ultralytics/nn/tasks.py,sha256=k6cl1H1hWtQXLxO_s5D6fa_DYyFzIekAMigh3lD36_A,42894
143
144
  ultralytics/nn/modules/__init__.py,sha256=Ga3MDpwX6DeI7VSH8joti5uleP4mgkQGolbe8RLZ2T8,2326
144
145
  ultralytics/nn/modules/block.py,sha256=n0ilrK3Vjdoc9-OwvY_Bp8Hw5N2BMHwgt-jfH4CzxFQ,25589
145
146
  ultralytics/nn/modules/conv.py,sha256=ndUYNL2f9DK41y1vVbtEusMByXy-LMMsBKlcWjRQ9Z8,12722
@@ -178,7 +179,7 @@ ultralytics/utils/plotting.py,sha256=YVJvEDozm1vm_Yf39jLPQ24Qst_f_lzEm-NeDjMElfQ
178
179
  ultralytics/utils/tal.py,sha256=xuIyryUjaaYHkHPG9GvBwh1xxN2Hq4y3hXOtuERehwY,16017
179
180
  ultralytics/utils/torch_utils.py,sha256=LgArI6s5wUhoDR0RBe3CoUkAkq3WpAg0YQPcL__rr_o,25182
180
181
  ultralytics/utils/triton.py,sha256=gg1finxno_tY2Ge9PMhmu7PI9wvoFZoiicdT4Bhqv3w,3936
181
- ultralytics/utils/tuner.py,sha256=mMa3PT5zvpHsTfKgOvFlRhDpogdCD1qSdNBVmU5Xop4,6003
182
+ ultralytics/utils/tuner.py,sha256=DTgMvg4kxCos-du5mqTGW5lPSXs-NF3YuAL0qpI5nQc,6111
182
183
  ultralytics/utils/callbacks/__init__.py,sha256=YrWqC3BVVaTLob4iCPR6I36mUxIUOpPJW7B_LjT78Qw,214
183
184
  ultralytics/utils/callbacks/base.py,sha256=sOe3JvyBFmRwVZ8_Q03u7JwTeOOm9CI4s9-UEhnG0xA,5777
184
185
  ultralytics/utils/callbacks/clearml.py,sha256=K7bDf5tS8xL4KeFMkoVDL2kKkil3f4qoKy8KfZkD854,5897
@@ -190,9 +191,9 @@ ultralytics/utils/callbacks/neptune.py,sha256=5Z3ua5YBTUS56FH8VQKQG1aaIo9fH8GEyz
190
191
  ultralytics/utils/callbacks/raytune.py,sha256=6OgGNuC35F29lw8Dl_d0lue4-iBR6dqrBVQnIRQDx4E,632
191
192
  ultralytics/utils/callbacks/tensorboard.py,sha256=hRmWjbqdA4RNaLuSZznuDcpOBW-_-_Ga0u-B8UU-7ZI,4134
192
193
  ultralytics/utils/callbacks/wb.py,sha256=4QI81nHdzgwhXHlmTiRxLqunvkKakLXYUhHTUY1ZeHA,6635
193
- ultralytics-8.1.31.dist-info/LICENSE,sha256=DZak_2itbUtvHzD3E7GNUYSRK6jdOJ-GqncQ2weavLA,34523
194
- ultralytics-8.1.31.dist-info/METADATA,sha256=e4f4Z8r6jRG5M_RoEfX0BDtwKwAx5S0HJEOLZDgZDZM,40330
195
- ultralytics-8.1.31.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
196
- ultralytics-8.1.31.dist-info/entry_points.txt,sha256=YM_wiKyTe9yRrsEfqvYolNO5ngwfoL4-NwgKzc8_7sI,93
197
- ultralytics-8.1.31.dist-info/top_level.txt,sha256=XP49TwiMw4QGsvTLSYiJhz1xF_k7ev5mQ8jJXaXi45Q,12
198
- ultralytics-8.1.31.dist-info/RECORD,,
194
+ ultralytics-8.1.32.dist-info/LICENSE,sha256=DZak_2itbUtvHzD3E7GNUYSRK6jdOJ-GqncQ2weavLA,34523
195
+ ultralytics-8.1.32.dist-info/METADATA,sha256=jQD4IWf9SXoTyZXKMyjFuNp8SrMPGdITqqNoQkRG-QI,40330
196
+ ultralytics-8.1.32.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
197
+ ultralytics-8.1.32.dist-info/entry_points.txt,sha256=YM_wiKyTe9yRrsEfqvYolNO5ngwfoL4-NwgKzc8_7sI,93
198
+ ultralytics-8.1.32.dist-info/top_level.txt,sha256=XP49TwiMw4QGsvTLSYiJhz1xF_k7ev5mQ8jJXaXi45Q,12
199
+ ultralytics-8.1.32.dist-info/RECORD,,