ultralytics 8.3.79__py3-none-any.whl → 8.3.80__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.
ultralytics/__init__.py CHANGED
@@ -1,6 +1,6 @@
1
1
  # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
2
2
 
3
- __version__ = "8.3.79"
3
+ __version__ = "8.3.80"
4
4
 
5
5
  import os
6
6
 
@@ -570,7 +570,7 @@ class Exporter:
570
570
  @try_export
571
571
  def export_openvino(self, prefix=colorstr("OpenVINO:")):
572
572
  """YOLO OpenVINO export."""
573
- check_requirements("openvino>=2024.5.0")
573
+ check_requirements("openvino>=2024.0.0,<2025.0.0")
574
574
  import openvino as ov
575
575
 
576
576
  LOGGER.info(f"\n{prefix} starting export with openvino {ov.__version__}...")
@@ -592,7 +592,7 @@ class Exporter:
592
592
  if self.model.task != "classify":
593
593
  ov_model.set_rt_info("fit_to_window_letterbox", ["model_info", "resize_type"])
594
594
 
595
- ov.runtime.save_model(ov_model, file, compress_to_fp16=self.args.half)
595
+ ov.save_model(ov_model, file, compress_to_fp16=self.args.half)
596
596
  yaml_save(Path(file).parent / "metadata.yaml", self.metadata) # add metadata.yaml
597
597
 
598
598
  if self.args.int8:
@@ -16,6 +16,7 @@ from pathlib import Path
16
16
  import torch
17
17
 
18
18
  from ultralytics.engine.model import Model
19
+ from ultralytics.utils import DEFAULT_CFG_DICT
19
20
  from ultralytics.utils.downloads import attempt_download_asset
20
21
  from ultralytics.utils.torch_utils import model_info
21
22
 
@@ -77,6 +78,7 @@ class NAS(Model):
77
78
  self.model.yaml = {} # for info()
78
79
  self.model.pt_path = weights # for export()
79
80
  self.model.task = "detect" # for export()
81
+ self.model.args = {**DEFAULT_CFG_DICT, **self.overrides} # for export()
80
82
 
81
83
  def info(self, detailed=False, verbose=True):
82
84
  """
@@ -244,7 +244,7 @@ class AutoBackend(nn.Module):
244
244
  # OpenVINO
245
245
  elif xml:
246
246
  LOGGER.info(f"Loading {w} for OpenVINO inference...")
247
- check_requirements("openvino>=2024.0.0")
247
+ check_requirements("openvino>=2024.0.0,<2025.0.0")
248
248
  import openvino as ov
249
249
 
250
250
  core = ov.Core()
@@ -600,7 +600,7 @@ class AutoBackend(nn.Module):
600
600
  results[userdata] = request.results
601
601
 
602
602
  # Create AsyncInferQueue, set the callback and start asynchronous inference for each input image
603
- async_queue = self.ov.runtime.AsyncInferQueue(self.ov_compiled_model)
603
+ async_queue = self.ov.AsyncInferQueue(self.ov_compiled_model)
604
604
  async_queue.set_callback(callback)
605
605
  for i in range(n):
606
606
  # Start async inference with userdata=i to specify the position in results list
ultralytics/utils/ops.py CHANGED
@@ -799,10 +799,11 @@ def regularize_rboxes(rboxes):
799
799
  (torch.Tensor): The regularized boxes.
800
800
  """
801
801
  x, y, w, h, t = rboxes.unbind(dim=-1)
802
- # Swap edge and angle if h >= w
803
- w_ = torch.where(w > h, w, h)
804
- h_ = torch.where(w > h, h, w)
805
- t = torch.where(w > h, t, t + math.pi / 2) % math.pi
802
+ # Swap edge if t >= pi/2 while not being symmetrically opposite
803
+ swap = t % math.pi >= math.pi / 2
804
+ w_ = torch.where(swap, h, w)
805
+ h_ = torch.where(swap, w, h)
806
+ t = t % (math.pi / 2)
806
807
  return torch.stack([x, y, w_, h_, t], dim=-1) # regularized boxes
807
808
 
808
809
 
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.2
2
2
  Name: ultralytics
3
- Version: 8.3.79
3
+ Version: 8.3.80
4
4
  Summary: Ultralytics YOLO 🚀 for SOTA object detection, multi-object tracking, instance segmentation, pose estimation and image classification.
5
5
  Author-email: Glenn Jocher <glenn.jocher@ultralytics.com>, Jing Qiu <jing.qiu@ultralytics.com>
6
6
  Maintainer-email: Ultralytics <hello@ultralytics.com>
@@ -63,7 +63,7 @@ Provides-Extra: export
63
63
  Requires-Dist: onnx>=1.12.0; extra == "export"
64
64
  Requires-Dist: coremltools>=7.0; (platform_system != "Windows" and python_version <= "3.11") and extra == "export"
65
65
  Requires-Dist: scikit-learn>=1.3.2; (platform_system != "Windows" and python_version <= "3.11") and extra == "export"
66
- Requires-Dist: openvino>=2024.0.0; extra == "export"
66
+ Requires-Dist: openvino<2025.0.0,>=2024.0.0; extra == "export"
67
67
  Requires-Dist: tensorflow>=2.0.0; extra == "export"
68
68
  Requires-Dist: tensorflowjs>=3.9.0; extra == "export"
69
69
  Requires-Dist: tensorstore>=0.1.63; (platform_machine == "aarch64" and python_version >= "3.9") and extra == "export"
@@ -7,7 +7,7 @@ tests/test_exports.py,sha256=T_z_NUS9URQXv83k5XNLHTuksJ8srtzbZnWuiiQWM98,9260
7
7
  tests/test_integrations.py,sha256=p3DMnnPMKsV0Qm82JVJUIY1UZ67xRgF9E8AaL76TEHE,6154
8
8
  tests/test_python.py,sha256=tW-EFJC2rjl_DvAa8khXGWYdypseQjrLjGHhe2p9r9A,23238
9
9
  tests/test_solutions.py,sha256=aY0G3vNzXGCENG9FD76MfUp7jgzeESPsUvbvQYBUvH0,4205
10
- ultralytics/__init__.py,sha256=moo1vmbr78zE56S4YpmOEngnomz8U-tJpyjDtGNeJxA,709
10
+ ultralytics/__init__.py,sha256=IAHoHNcj5aqG2zLWJQS55f1IkUvdaho1wEs4FE2yBQ0,709
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=qP44HnFP4QcC5FQz29A-EGTuwdtxXAzPvw_IvCVmiqA,39771
@@ -107,7 +107,7 @@ ultralytics/data/loaders.py,sha256=JOwXbz-dxgG2bx0_cQHp-olz5FleoCX8EzrUvZ77vvg,2
107
107
  ultralytics/data/split_dota.py,sha256=YI-i2MqdiBt06W67TJnBXQHJrqTnkJDJ3zzoL0UZVro,10733
108
108
  ultralytics/data/utils.py,sha256=5YMU5396oAFPwTy2y0MCU2WipF6Rt-7xNtmHKRCA4fI,33838
109
109
  ultralytics/engine/__init__.py,sha256=lm6MckFYCPTbqIoX7w0s_daxdjNeBeKW6DXppv1-QUM,70
110
- ultralytics/engine/exporter.py,sha256=qeEV-w8cdYuVJP4RaKbq46xJmsYGUFJnkplfEUCIm_c,77013
110
+ ultralytics/engine/exporter.py,sha256=DslfLk7H44ly4-kCY0u6SLSRjdx7NO5lzfnnD1_c7wI,77015
111
111
  ultralytics/engine/model.py,sha256=s8HsSBvdRgSbnKGULr7YW-ZWJKJsQpOoHd9Aih_nMt0,53427
112
112
  ultralytics/engine/predictor.py,sha256=jiYDAjupOlRUpPvw9tu7or9PjXtLm-YCRiawANtWxj0,17881
113
113
  ultralytics/engine/results.py,sha256=hWlO2e58BPUJ5R4Jl4iirBPaZ8BypcNu_cNQ2NHpUqM,78111
@@ -126,7 +126,7 @@ ultralytics/models/fastsam/predict.py,sha256=IqdetKBwkrrLnUWRf37vjiNFkudqVd_OPEw
126
126
  ultralytics/models/fastsam/utils.py,sha256=Sl6vXHzK3G6SD-NdxmsiiHM4chlyaHl0pjSkU3Wb3UU,742
127
127
  ultralytics/models/fastsam/val.py,sha256=Dc2X2bOF8rAIDN1eXLOodKPy3YpCVWAnevt7OhTozok,1995
128
128
  ultralytics/models/nas/__init__.py,sha256=wybeHZuAXMNeXMjKTbK55FZmXJkA4K9IozDeFM9OB-s,207
129
- ultralytics/models/nas/model.py,sha256=93bmemeFxe0Xbj3VrNf6EIfgiJZJMsg2u8tWajxh47c,3262
129
+ ultralytics/models/nas/model.py,sha256=slVlIe-44BS6L1AJkfyin4E2V5aBQEkBBd6nRAXcAoQ,3390
130
130
  ultralytics/models/nas/predict.py,sha256=nzVGTdUb0E_IjmWksX_T61q80hbrjEovihTzTJ1rfmA,2124
131
131
  ultralytics/models/nas/val.py,sha256=CSqmcuAcuJ5SQ7mo364RdXLGeu2XATyRY8Z84VGGX5o,1497
132
132
  ultralytics/models/rtdetr/__init__.py,sha256=_jEHmOjI_QP_nT3XJXLgYHQ6bXG4EL8Gnvn1y_eev1g,225
@@ -177,7 +177,7 @@ ultralytics/models/yolo/world/__init__.py,sha256=nlh8I6t8hMGz_vZg8QSlsUW1R-2eKvn
177
177
  ultralytics/models/yolo/world/train.py,sha256=6PVmQ0G-22OOPPwP_rqSobe2LM6e2b_lC7lJCdW3UIk,3714
178
178
  ultralytics/models/yolo/world/train_world.py,sha256=sCtg4Hnq9Y7amYjlQsdvTHXH8cKSooipvcXu_1Iyb2k,4885
179
179
  ultralytics/nn/__init__.py,sha256=rjociYD9lo_K-d-1s6TbdWklPLjTcEHk7OIlRDJstIE,615
180
- ultralytics/nn/autobackend.py,sha256=uVLr4GymWtz4D86QXzmqs7LwoLjv3VycFLfdKq9rG1U,37377
180
+ ultralytics/nn/autobackend.py,sha256=WyvIsfba4eN4tAtvuuy089-UDLpsvXfO4fTBdPnUPP8,37379
181
181
  ultralytics/nn/tasks.py,sha256=rKyTShwk1RtWBnbHObcSamxXoCUiwzl0K5QFYaw56Hw,49030
182
182
  ultralytics/nn/modules/__init__.py,sha256=pVV5SSu6ktOusdVFr1kHK_WOkVLjCLO2W5XaLH-NF8w,2737
183
183
  ultralytics/nn/modules/activation.py,sha256=oRkhMdqlNpIxQb35pTSUeHV-h0VyLl96GOqvIZ4OvT8,923
@@ -220,7 +220,7 @@ ultralytics/utils/files.py,sha256=c85NRofjGPMcpkV-yUo1Cwk8ZVquBGCEKlzbSVtXkQA,82
220
220
  ultralytics/utils/instance.py,sha256=z1oyyvz7wnCSUW_bvi0TbgAL0VxJtAWWXV9KWCoyJ_k,16887
221
221
  ultralytics/utils/loss.py,sha256=paRY8K7R4pcUGJfApVzZx-m_iFzzMbHm5GgiaixfDuU,34179
222
222
  ultralytics/utils/metrics.py,sha256=M15LVYzTGgmahkALKwKU3iYDoJIZ3M4824FLqsJ9qeU,54224
223
- ultralytics/utils/ops.py,sha256=izQr5GvgzmaD-GXaqxIjLE525JnvgLetOtuq_EOaxM8,34584
223
+ ultralytics/utils/ops.py,sha256=e7HNeufSrOnaHaie8w-QHNnyaClcHuiGZvr3CXRABsU,34621
224
224
  ultralytics/utils/patches.py,sha256=ARR89dP4YKq7Dd3g2eU-ukbnc2lo3BELukL_1c_d854,3298
225
225
  ultralytics/utils/plotting.py,sha256=hKji4TyxAmCXdSL264VX6dsC2AZYiL9StShI02dcAOM,62990
226
226
  ultralytics/utils/tal.py,sha256=DO-c006HEI62pcrNRpmt4lpqJPC5yu3veRDOvUuExno,18498
@@ -238,9 +238,9 @@ ultralytics/utils/callbacks/neptune.py,sha256=waZ_bRu0-qBKujTLuqonC2gx2DkgBuVnfq
238
238
  ultralytics/utils/callbacks/raytune.py,sha256=A_NVWjyPNf2m6iB-mbW7SMpyqM9QBvpbPa-MCMFMtdk,727
239
239
  ultralytics/utils/callbacks/tensorboard.py,sha256=JHOEVlNQ5dYJPd4Z-EvqbXowuK5uA0p8wPgyyaIUQs0,4194
240
240
  ultralytics/utils/callbacks/wb.py,sha256=ayhT2y62AcSOacnawshATU0rWrlSFQ77mrGgBdRl3W4,7086
241
- ultralytics-8.3.79.dist-info/LICENSE,sha256=DZak_2itbUtvHzD3E7GNUYSRK6jdOJ-GqncQ2weavLA,34523
242
- ultralytics-8.3.79.dist-info/METADATA,sha256=Qg8FyAGAg92474miDCJhNUdGM-siQ8ypdixaKpzYyko,35158
243
- ultralytics-8.3.79.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
244
- ultralytics-8.3.79.dist-info/entry_points.txt,sha256=YM_wiKyTe9yRrsEfqvYolNO5ngwfoL4-NwgKzc8_7sI,93
245
- ultralytics-8.3.79.dist-info/top_level.txt,sha256=XP49TwiMw4QGsvTLSYiJhz1xF_k7ev5mQ8jJXaXi45Q,12
246
- ultralytics-8.3.79.dist-info/RECORD,,
241
+ ultralytics-8.3.80.dist-info/LICENSE,sha256=DZak_2itbUtvHzD3E7GNUYSRK6jdOJ-GqncQ2weavLA,34523
242
+ ultralytics-8.3.80.dist-info/METADATA,sha256=IPuK8BOShhagOgRfZLSiJMjRHY-X0dYYmBnVil7FZu8,35168
243
+ ultralytics-8.3.80.dist-info/WHEEL,sha256=nn6H5-ilmfVryoAQl3ZQ2l8SH5imPWFpm1A5FgEuFV4,91
244
+ ultralytics-8.3.80.dist-info/entry_points.txt,sha256=YM_wiKyTe9yRrsEfqvYolNO5ngwfoL4-NwgKzc8_7sI,93
245
+ ultralytics-8.3.80.dist-info/top_level.txt,sha256=XP49TwiMw4QGsvTLSYiJhz1xF_k7ev5mQ8jJXaXi45Q,12
246
+ ultralytics-8.3.80.dist-info/RECORD,,
@@ -1,5 +1,5 @@
1
1
  Wheel-Version: 1.0
2
- Generator: setuptools (75.8.0)
2
+ Generator: setuptools (75.8.1)
3
3
  Root-Is-Purelib: true
4
4
  Tag: py3-none-any
5
5