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 +1 -1
- ultralytics/cfg/__init__.py +1 -1
- ultralytics/cfg/datasets/african-wildlife.yaml +24 -0
- ultralytics/nn/tasks.py +1 -1
- ultralytics/utils/tuner.py +4 -3
- {ultralytics-8.1.31.dist-info → ultralytics-8.1.32.dist-info}/METADATA +1 -1
- {ultralytics-8.1.31.dist-info → ultralytics-8.1.32.dist-info}/RECORD +11 -10
- {ultralytics-8.1.31.dist-info → ultralytics-8.1.32.dist-info}/LICENSE +0 -0
- {ultralytics-8.1.31.dist-info → ultralytics-8.1.32.dist-info}/WHEEL +0 -0
- {ultralytics-8.1.31.dist-info → ultralytics-8.1.32.dist-info}/entry_points.txt +0 -0
- {ultralytics-8.1.31.dist-info → ultralytics-8.1.32.dist-info}/top_level.txt +0 -0
ultralytics/__init__.py
CHANGED
ultralytics/cfg/__init__.py
CHANGED
|
@@ -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
|
|
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)
|
ultralytics/utils/tuner.py
CHANGED
|
@@ -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('
|
|
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.
|
|
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=
|
|
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=
|
|
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=
|
|
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=
|
|
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.
|
|
194
|
-
ultralytics-8.1.
|
|
195
|
-
ultralytics-8.1.
|
|
196
|
-
ultralytics-8.1.
|
|
197
|
-
ultralytics-8.1.
|
|
198
|
-
ultralytics-8.1.
|
|
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,,
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|