ultralytics 8.2.53__py3-none-any.whl → 8.2.55__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.2.53"
3
+ __version__ = "8.2.55"
4
4
 
5
5
  import os
6
6
 
@@ -544,8 +544,9 @@ def get_best_youtube_url(url, method="pytube"):
544
544
  (str): The URL of the best quality MP4 video stream, or None if no suitable stream is found.
545
545
  """
546
546
  if method == "pytube":
547
- check_requirements("pytube")
548
- from pytube import YouTube
547
+ # Switched from pytube to pytubefix to resolve https://github.com/pytube/pytube/issues/1954
548
+ check_requirements("pytubefix")
549
+ from pytubefix import YouTube
549
550
 
550
551
  streams = YouTube(url).streams.filter(file_extension="mp4", only_video=True)
551
552
  streams = sorted(streams, key=lambda s: s.resolution, reverse=True) # sort streams by resolution
@@ -1017,13 +1017,13 @@ class Exporter:
1017
1017
  """Add metadata to *.tflite models per https://www.tensorflow.org/lite/models/convert/metadata."""
1018
1018
  import flatbuffers
1019
1019
 
1020
- if ARM64:
1021
- from tflite_support import metadata # noqa
1022
- from tflite_support import metadata_schema_py_generated as schema # noqa
1023
- else:
1020
+ try:
1024
1021
  # TFLite Support bug https://github.com/tensorflow/tflite-support/issues/954#issuecomment-2108570845
1025
1022
  from tensorflow_lite_support.metadata import metadata_schema_py_generated as schema # noqa
1026
1023
  from tensorflow_lite_support.metadata.python import metadata # noqa
1024
+ except ImportError: # ARM64 systems may not have the 'tensorflow_lite_support' package available
1025
+ from tflite_support import metadata # noqa
1026
+ from tflite_support import metadata_schema_py_generated as schema # noqa
1027
1027
 
1028
1028
  # Create model info
1029
1029
  model_meta = schema.ModelMetadataT()
ultralytics/hub/utils.py CHANGED
@@ -185,7 +185,7 @@ class Events:
185
185
  def __init__(self):
186
186
  """Initializes the Events object with default values for events, rate_limit, and metadata."""
187
187
  self.events = [] # events list
188
- self.rate_limit = 60.0 # rate limit (seconds)
188
+ self.rate_limit = 30.0 # rate limit (seconds)
189
189
  self.t = 0.0 # rate limit timer (seconds)
190
190
  self.metadata = {
191
191
  "cli": Path(ARGV[0]).name == "yolo",
@@ -17,7 +17,7 @@ from ultralytics.utils import LOGGER, TQDM, checks, clean_url, emojis, is_online
17
17
  # Define Ultralytics GitHub assets maintained at https://github.com/ultralytics/assets
18
18
  GITHUB_ASSETS_REPO = "ultralytics/assets"
19
19
  GITHUB_ASSETS_NAMES = (
20
- [f"yolov8{k}{suffix}.pt" for k in "nsmlx" for suffix in ("", "-cls", "-seg", "-pose", "-obb")]
20
+ [f"yolov8{k}{suffix}.pt" for k in "nsmlx" for suffix in ("", "-cls", "-seg", "-pose", "-obb", "-oiv7")]
21
21
  + [f"yolov5{k}{resolution}u.pt" for k in "nsmlx" for resolution in ("", "6")]
22
22
  + [f"yolov3{k}u.pt" for k in ("", "-spp", "-tiny")]
23
23
  + [f"yolov8{k}-world.pt" for k in "smlx"]
@@ -271,7 +271,7 @@ def model_info(model, detailed=False, verbose=True, imgsz=640):
271
271
  fs = f", {flops:.1f} GFLOPs" if flops else ""
272
272
  yaml_file = getattr(model, "yaml_file", "") or getattr(model, "yaml", {}).get("yaml_file", "")
273
273
  model_name = Path(yaml_file).stem.replace("yolo", "YOLO") or "Model"
274
- LOGGER.info(f"{model_name} summary{fused}: {n_l} layers, {n_p} parameters, {n_g} gradients{fs}")
274
+ LOGGER.info(f"{model_name} summary{fused}: {n_l:,} layers, {n_p:,} parameters, {n_g:,} gradients{fs}")
275
275
  return n_l, n_p, n_g, flops
276
276
 
277
277
 
@@ -513,6 +513,9 @@ def strip_optimizer(f: Union[str, Path] = "best.pt", s: str = "") -> None:
513
513
  for f in Path('path/to/model/checkpoints').rglob('*.pt'):
514
514
  strip_optimizer(f)
515
515
  ```
516
+
517
+ Note:
518
+ Use `ultralytics.nn.torch_safe_load` for missing modules with `x = torch_safe_load(f)[0]`
516
519
  """
517
520
  try:
518
521
  x = torch.load(f, map_location=torch.device("cpu"))
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: ultralytics
3
- Version: 8.2.53
3
+ Version: 8.2.55
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
@@ -7,7 +7,7 @@ tests/test_explorer.py,sha256=NcxSJeB6FxwkN09hQl7nnQL--HjfHB_WcZk0mEmBNHI,2215
7
7
  tests/test_exports.py,sha256=Uezf3OatpPHlo5qoPw-2kqkZxuMCF9L4XF2riD4vmII,8225
8
8
  tests/test_integrations.py,sha256=xglcfMPjfVh346PV8WTpk6tBxraCXEFJEQyyJMr5tyU,6064
9
9
  tests/test_python.py,sha256=qhtSQ7NDfBChsVUxeSwfUIkoKq0S1Z-Rd9_MP023Y5k,21794
10
- ultralytics/__init__.py,sha256=fDa5shd9ITuM3SOYAGwCfW_q3NMC2eqOxGXOpgnr4iY,694
10
+ ultralytics/__init__.py,sha256=Y1BrkAwy_qxD_akh7Rx6a1_h5y4ghtjTj48jCA4e8OQ,694
11
11
  ultralytics/assets/bus.jpg,sha256=wCAZxJecGR63Od3ZRERe9Aja1Weayrb9Ug751DS_vGM,137419
12
12
  ultralytics/assets/zidane.jpg,sha256=Ftc4aeMmen1O0A3o6GCDO9FlfBslLpTAw0gnetx7bts,50427
13
13
  ultralytics/cfg/__init__.py,sha256=MqUsV-Mdk80dO64yY7JmplEO0Awb-25Lfx4YC9QYxhc,26210
@@ -88,7 +88,7 @@ ultralytics/data/base.py,sha256=C3teLnw97ZTbpJHT9P7yYWosAKocMzgJjRe1rxgfpls,1352
88
88
  ultralytics/data/build.py,sha256=AfMmz0sHIYmwry_90tEJFRk_kz0S3SolScVXqYHiT08,7261
89
89
  ultralytics/data/converter.py,sha256=7640xKuf7LPeoTwoCvgbIXM5xbzyq72Hu2Rf2lrgjRY,17554
90
90
  ultralytics/data/dataset.py,sha256=q8g8cUAabdhqahL0a0cbulOey29UFXY51VDEFJN_x0c,22444
91
- ultralytics/data/loaders.py,sha256=UUL7yOmuseAG5RBVI-kLrLr42Vm4kL05Qqnc5jAmNW0,23972
91
+ ultralytics/data/loaders.py,sha256=XnwJsrejnigaG0wwivKccFUxq002czYa4cgVfGzsFms,24078
92
92
  ultralytics/data/split_dota.py,sha256=fWezt1Bo3jiZ6AyUWdBtTUuvLamPv1t7JD-DirM9gQ8,10142
93
93
  ultralytics/data/utils.py,sha256=GHmqx6e5yRfcUD2Qkwk-tQfhXCwtUMFD3Uf6d699nGo,31046
94
94
  ultralytics/data/explorer/__init__.py,sha256=-Y3m1ZedepOQUv_KW82zaGxvU_PSHcuwUTFqG9BhAr4,113
@@ -97,7 +97,7 @@ ultralytics/data/explorer/utils.py,sha256=EvvukQiQUTBrsZznmMnyEX2EqTuwZo_Geyc8yf
97
97
  ultralytics/data/explorer/gui/__init__.py,sha256=mHtJuK4hwF8cuV-VHDc7tp6u6D1gHz2Z7JI8grmQDTs,42
98
98
  ultralytics/data/explorer/gui/dash.py,sha256=CPlFIIhf53j_YVAqealsC3AbcztdPqZxfniQcBnlKK4,10042
99
99
  ultralytics/engine/__init__.py,sha256=mHtJuK4hwF8cuV-VHDc7tp6u6D1gHz2Z7JI8grmQDTs,42
100
- ultralytics/engine/exporter.py,sha256=-qidFlOXEnzOJR8c4YhsZqugonN0tpyk_Ye4dIrTsT4,58642
100
+ ultralytics/engine/exporter.py,sha256=yV5DKjz5DZ6BrW8mOC5Nb5eDcuCc93Ft-RQwJ21xVZs,58729
101
101
  ultralytics/engine/model.py,sha256=8qD5irabp8BF7bBZGwztCu8yAVQQp1kksYSea9EhdEo,39078
102
102
  ultralytics/engine/predictor.py,sha256=W58kDCFH2AfoFzpGbos3k8zUEVsLunBuM8sc2B64rPY,17449
103
103
  ultralytics/engine/results.py,sha256=5MevvBz0E-cpDf55FqweInlKdcQPb7sz0EgZSROJqw4,35817
@@ -107,7 +107,7 @@ ultralytics/engine/validator.py,sha256=Y21Uo8_Zto4qjk_YqQk6k7tyfpq_Qk9cfjeXeyDRx
107
107
  ultralytics/hub/__init__.py,sha256=93bqI8x8-MfDYdKkQVduuocUiQj3WGnk1nIk0li08zA,5663
108
108
  ultralytics/hub/auth.py,sha256=FID58NE6fh7Op_B45QOpWBw1qoBN0ponL16uvyb2dZ8,5399
109
109
  ultralytics/hub/session.py,sha256=uXkP8AayJClLUD9TP8AlJSqxm-OmTgCmTXl1TkO6jQc,16147
110
- ultralytics/hub/utils.py,sha256=RpFDFp9biUK70Mswzz2o3uEu4xwQxRaStPS19U2gu0g,9721
110
+ ultralytics/hub/utils.py,sha256=tXfM3QbXBcf4Y6StgHI1pktT4OM7Ic9eF3xiBFHGlhY,9721
111
111
  ultralytics/models/__init__.py,sha256=TT9iLCL_n9Y80dcUq0Fo-p-GRZCSU2vrWXM3CoMwqqE,265
112
112
  ultralytics/models/fastsam/__init__.py,sha256=0dt65jZ_5b7Q-mdXN8MSEkgnFRA0FIwlel_LS2RaOlU,254
113
113
  ultralytics/models/fastsam/model.py,sha256=c7GGwaa9AXssJFwrcuytFHpPOlgSrS3n0utyf4JSL2o,1055
@@ -196,7 +196,7 @@ ultralytics/utils/autobatch.py,sha256=gPFcREMsMHRAuTQiBnNZ9Mm1XNqmQW-uMPhveDFEQ_
196
196
  ultralytics/utils/benchmarks.py,sha256=EudI2wNzOPPNjf9J0AiROe32yD4G8nTg5tcZOZSJSw8,23581
197
197
  ultralytics/utils/checks.py,sha256=QIltfNxlZdMOTzXqU815MBIevMj_TKBU_VeVXqjXdOo,28411
198
198
  ultralytics/utils/dist.py,sha256=NDFga-uKxkBX2zLxFHSene_cCiGQJoyOeCXcN9JIOIk,2358
199
- ultralytics/utils/downloads.py,sha256=MPuCfS7zZh_uco0Ddqh6NdKqdmUXXpk0Wz_GS99ltlc,21920
199
+ ultralytics/utils/downloads.py,sha256=60GL4gZ3kIHvu-8_PrPY1WBNuXPMvL5si-6vCX_qbQ4,21929
200
200
  ultralytics/utils/errors.py,sha256=GqP_Jgj_n0paxn8OMhn3DTCgoNkB2WjUcUaqs-M6SQk,816
201
201
  ultralytics/utils/files.py,sha256=TVfY0Wi5IsUc4YdsDzC0dAg-jAP5exYvwqB3VmXhDLY,6761
202
202
  ultralytics/utils/instance.py,sha256=5daM5nkxBv9hr5QzyII8zmuFj24hHuNtcr4EMCHAtpY,15654
@@ -206,7 +206,7 @@ ultralytics/utils/ops.py,sha256=Jlb0YBkN_SMVT2AjKPEjxgOtgnj7i7HTBh9FEwpoprU,3350
206
206
  ultralytics/utils/patches.py,sha256=SgMqeMsq2K6JoBJP1NplXMl9C6rK0JeJUChjBrJOneo,2750
207
207
  ultralytics/utils/plotting.py,sha256=icSUqsmJLpeXyVAIt8vxpbrxTe40mwiF5ay4el3IXl0,55584
208
208
  ultralytics/utils/tal.py,sha256=xuIyryUjaaYHkHPG9GvBwh1xxN2Hq4y3hXOtuERehwY,16017
209
- ultralytics/utils/torch_utils.py,sha256=uuiXENrjF8a0PydZRfdp3bQ4oQZ9FyERXXfqGyXLCg0,27713
209
+ ultralytics/utils/torch_utils.py,sha256=EqBLg_G4x31InrTEvUvvMyxWaFaZ7UNts0tUUQsQmLY,27828
210
210
  ultralytics/utils/triton.py,sha256=gg1finxno_tY2Ge9PMhmu7PI9wvoFZoiicdT4Bhqv3w,3936
211
211
  ultralytics/utils/tuner.py,sha256=49KAadKZsUeCpwIm5Sn0grb0RPcMNI8vHGLwroDEJNI,6171
212
212
  ultralytics/utils/callbacks/__init__.py,sha256=YrWqC3BVVaTLob4iCPR6I36mUxIUOpPJW7B_LjT78Qw,214
@@ -220,9 +220,9 @@ ultralytics/utils/callbacks/neptune.py,sha256=5Z3ua5YBTUS56FH8VQKQG1aaIo9fH8GEyz
220
220
  ultralytics/utils/callbacks/raytune.py,sha256=ODVYzy-CoM4Uge0zjkh3Hnh9nF2M0vhDrSenXnvcizw,705
221
221
  ultralytics/utils/callbacks/tensorboard.py,sha256=QEgOVhUqY9akOs5TJIwz1Rvn6l32xWLpOxlwEyWF0B8,4136
222
222
  ultralytics/utils/callbacks/wb.py,sha256=9-fjQIdLjr3b73DTE3rHO171KvbH1VweJ-bmbv-rqTw,6747
223
- ultralytics-8.2.53.dist-info/LICENSE,sha256=DZak_2itbUtvHzD3E7GNUYSRK6jdOJ-GqncQ2weavLA,34523
224
- ultralytics-8.2.53.dist-info/METADATA,sha256=9jxWIong0zNd04Gd2rvhQGDx7oYTgbMkrw5jbBCgsVY,41217
225
- ultralytics-8.2.53.dist-info/WHEEL,sha256=Z4pYXqR_rTB7OWNDYFOm1qRk0RX6GFP2o8LgvP453Hk,91
226
- ultralytics-8.2.53.dist-info/entry_points.txt,sha256=YM_wiKyTe9yRrsEfqvYolNO5ngwfoL4-NwgKzc8_7sI,93
227
- ultralytics-8.2.53.dist-info/top_level.txt,sha256=XP49TwiMw4QGsvTLSYiJhz1xF_k7ev5mQ8jJXaXi45Q,12
228
- ultralytics-8.2.53.dist-info/RECORD,,
223
+ ultralytics-8.2.55.dist-info/LICENSE,sha256=DZak_2itbUtvHzD3E7GNUYSRK6jdOJ-GqncQ2weavLA,34523
224
+ ultralytics-8.2.55.dist-info/METADATA,sha256=3aZwlCXP2FlgcO8Ht-MvPW9ymhAbh_nFPJgC_nO8esk,41217
225
+ ultralytics-8.2.55.dist-info/WHEEL,sha256=Z4pYXqR_rTB7OWNDYFOm1qRk0RX6GFP2o8LgvP453Hk,91
226
+ ultralytics-8.2.55.dist-info/entry_points.txt,sha256=YM_wiKyTe9yRrsEfqvYolNO5ngwfoL4-NwgKzc8_7sI,93
227
+ ultralytics-8.2.55.dist-info/top_level.txt,sha256=XP49TwiMw4QGsvTLSYiJhz1xF_k7ev5mQ8jJXaXi45Q,12
228
+ ultralytics-8.2.55.dist-info/RECORD,,