ultralytics 8.3.18__py3-none-any.whl → 8.3.20__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 YOLO 🚀, AGPL-3.0 license
2
2
 
3
- __version__ = "8.3.18"
3
+ __version__ = "8.3.20"
4
4
 
5
5
  import os
6
6
 
@@ -691,10 +691,10 @@ class Exporter:
691
691
  import tensorrt as trt # noqa
692
692
  except ImportError:
693
693
  if LINUX:
694
- check_requirements("tensorrt>7.0.0,<=10.1.0")
694
+ check_requirements("tensorrt>7.0.0,!=10.1.0")
695
695
  import tensorrt as trt # noqa
696
696
  check_version(trt.__version__, ">=7.0.0", hard=True)
697
- check_version(trt.__version__, "<=10.1.0", msg="https://github.com/ultralytics/ultralytics/pull/14239")
697
+ check_version(trt.__version__, "!=10.1.0", msg="https://github.com/ultralytics/ultralytics/pull/14239")
698
698
 
699
699
  # Setup and checks
700
700
  LOGGER.info(f"\n{prefix} starting export with TensorRT {trt.__version__}...")
@@ -226,10 +226,10 @@ class AutoBackend(nn.Module):
226
226
  import tensorrt as trt # noqa https://developer.nvidia.com/nvidia-tensorrt-download
227
227
  except ImportError:
228
228
  if LINUX:
229
- check_requirements("tensorrt>7.0.0,<=10.1.0")
229
+ check_requirements("tensorrt>7.0.0,!=10.1.0")
230
230
  import tensorrt as trt # noqa
231
231
  check_version(trt.__version__, ">=7.0.0", hard=True)
232
- check_version(trt.__version__, "<=10.1.0", msg="https://github.com/ultralytics/ultralytics/pull/14239")
232
+ check_version(trt.__version__, "!=10.1.0", msg="https://github.com/ultralytics/ultralytics/pull/14239")
233
233
  if device.type == "cpu":
234
234
  device = torch.device("cuda:0")
235
235
  Binding = namedtuple("Binding", ("name", "dtype", "shape", "data", "ptr"))
@@ -137,17 +137,19 @@ def on_train_end(trainer):
137
137
  if trainer.best.exists():
138
138
  art.add_file(trainer.best)
139
139
  wb.run.log_artifact(art, aliases=["best"])
140
- for curve_name, curve_values in zip(trainer.validator.metrics.curves, trainer.validator.metrics.curves_results):
141
- x, y, x_title, y_title = curve_values
142
- _plot_curve(
143
- x,
144
- y,
145
- names=list(trainer.validator.metrics.names.values()),
146
- id=f"curves/{curve_name}",
147
- title=curve_name,
148
- x_title=x_title,
149
- y_title=y_title,
150
- )
140
+ # Check if we actually have plots to save
141
+ if trainer.args.plots:
142
+ for curve_name, curve_values in zip(trainer.validator.metrics.curves, trainer.validator.metrics.curves_results):
143
+ x, y, x_title, y_title = curve_values
144
+ _plot_curve(
145
+ x,
146
+ y,
147
+ names=list(trainer.validator.metrics.names.values()),
148
+ id=f"curves/{curve_name}",
149
+ title=curve_name,
150
+ x_title=x_title,
151
+ y_title=y_title,
152
+ )
151
153
  wb.run.finish() # required or run continues on dashboard
152
154
 
153
155
 
@@ -688,7 +688,7 @@ def check_amp(model):
688
688
 
689
689
  im = ASSETS / "bus.jpg" # image to check
690
690
  prefix = colorstr("AMP: ")
691
- LOGGER.info(f"{prefix}running Automatic Mixed Precision (AMP) checks with YOLO11n...")
691
+ LOGGER.info(f"{prefix}running Automatic Mixed Precision (AMP) checks...")
692
692
  warning_msg = "Setting 'amp=True'. If you experience zero-mAP or NaN losses you can disable AMP with amp=False."
693
693
  try:
694
694
  from ultralytics import YOLO
@@ -696,11 +696,13 @@ def check_amp(model):
696
696
  assert amp_allclose(YOLO("yolo11n.pt"), im)
697
697
  LOGGER.info(f"{prefix}checks passed ✅")
698
698
  except ConnectionError:
699
- LOGGER.warning(f"{prefix}checks skipped ⚠️, offline and unable to download YOLO11n. {warning_msg}")
699
+ LOGGER.warning(
700
+ f"{prefix}checks skipped ⚠️. " f"Offline and unable to download YOLO11n for AMP checks. {warning_msg}"
701
+ )
700
702
  except (AttributeError, ModuleNotFoundError):
701
703
  LOGGER.warning(
702
704
  f"{prefix}checks skipped ⚠️. "
703
- f"Unable to load YOLO11n due to possible Ultralytics package modifications. {warning_msg}"
705
+ f"Unable to load YOLO11n for AMP checks due to possible Ultralytics package modifications. {warning_msg}"
704
706
  )
705
707
  except AssertionError:
706
708
  LOGGER.warning(
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: ultralytics
3
- Version: 8.3.18
3
+ Version: 8.3.20
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>
@@ -7,7 +7,7 @@ tests/test_exports.py,sha256=fpTKEVBUGLF3WiZPNKRs-IEcIY4cfxgvgKjUNfodjww,8042
7
7
  tests/test_integrations.py,sha256=f5-QCUk1SU_-qn4mBCZwS3GN3tXEBIIXo4z2EhExbHw,6126
8
8
  tests/test_python.py,sha256=I1RRdCwLdrc3jX06huVxct8HX8ccQOmQgVpuEflRl0U,23560
9
9
  tests/test_solutions.py,sha256=sPYhy2d814mIVvojQeVxeZPu0IVy01_Y8zuMcu_9GF0,3790
10
- ultralytics/__init__.py,sha256=qf4JhbZAMoae9nuZSVyF_RxYv_myNq4dkkvb8c5PuEo,681
10
+ ultralytics/__init__.py,sha256=LOaTgFJqB_O9Bq-7ibpqNKtbRN2K8BoMduHXgt5gFa4,681
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=hj7JH0TgnF9OlKi4uJBUu-KeQh4PFnSQCTiD_h_u_Sg,32399
@@ -99,7 +99,7 @@ ultralytics/data/loaders.py,sha256=Fr70Q9p9t7buLW_8R2_lI_nyCMG033gWSxvwy1M-a-U,2
99
99
  ultralytics/data/split_dota.py,sha256=eFafJ7Vg52wj6KDCHFJAf1tKzyPD5YaPB8kM4VX5Aeg,10688
100
100
  ultralytics/data/utils.py,sha256=bmWEIrdogj4kssZQSJdSbIF8QsJU00lo-EY-Mgcqv4M,31073
101
101
  ultralytics/engine/__init__.py,sha256=mHtJuK4hwF8cuV-VHDc7tp6u6D1gHz2Z7JI8grmQDTs,42
102
- ultralytics/engine/exporter.py,sha256=OQONIGMLBKgkhfUC4CV7mRWfyo_VV03SA5SnaetBIsM,57662
102
+ ultralytics/engine/exporter.py,sha256=EF7qoga6PSFtsG27D6Q4zUnbbu-aoKSLtkJtUmkPx2w,57662
103
103
  ultralytics/engine/model.py,sha256=pvL1uf-wwdWL8Iph7VEAYn1-z7wEHzVug21V_0_gO6M,51456
104
104
  ultralytics/engine/predictor.py,sha256=keTelEeo23Dcbs-XvmRWAPIs4pbCNDtsMBz88WM1eK8,17534
105
105
  ultralytics/engine/results.py,sha256=BxanBI8PhBCfs-9cSy-GS6naScuiD3hdvUAJWPW2mS0,75043
@@ -169,7 +169,7 @@ ultralytics/models/yolo/world/__init__.py,sha256=3VTH0q4NOt2EWRom15yCymvmvm0Etp2
169
169
  ultralytics/models/yolo/world/train.py,sha256=gaDrAmLJpg9qDtmL5evA5HsV2yb4RTRSfk2EDYrHdRg,3686
170
170
  ultralytics/models/yolo/world/train_world.py,sha256=IsnCEVt6DcM9lUskCKmIN-M8MM79xLpwTRqRoAHUnZ4,4857
171
171
  ultralytics/nn/__init__.py,sha256=4BPLHY89xEM_al5uK0aOmFgiML6CMGEZbezxOvTjOEs,587
172
- ultralytics/nn/autobackend.py,sha256=sFo9vx3y1M3lzaROMvMFfar7EngEn4BF5-_439r_eZA,31798
172
+ ultralytics/nn/autobackend.py,sha256=WGYfQtJ6GpbYTFSOFTpL7-bo4Z2qlr_G1xvH56LZRZU,31798
173
173
  ultralytics/nn/tasks.py,sha256=NWe0cL7A0LpsP3S1Efvi2NutAjWc_rGYMJMwAeb2bAg,48605
174
174
  ultralytics/nn/modules/__init__.py,sha256=xhW2BennT9U_VaMXVpRu-bdLgp1BXt9L8mkIUBE3idU,2625
175
175
  ultralytics/nn/modules/activation.py,sha256=chhn469wnRHEs5BMGNBYXwPYZc_7-urspTT8fnBd-xA,895
@@ -201,7 +201,7 @@ ultralytics/trackers/utils/matching.py,sha256=3Ie1WNNRZ4_q3365F03XD7Nr9juZB_08mw
201
201
  ultralytics/utils/__init__.py,sha256=YENakCDRXzHXy3e_MzTd2iCSmNVcGzCUgGhdiNGpaV4,48690
202
202
  ultralytics/utils/autobatch.py,sha256=BO9MCRtrLDtrDQaxqV0BdjaYsgXf-q07Y3_VdGp4URY,4330
203
203
  ultralytics/utils/benchmarks.py,sha256=R3_jtwLd48azPSXmtIhqlzduUvflk0FOY8GJOJ6mB6E,25097
204
- ultralytics/utils/checks.py,sha256=jjCIAT6yBQ2v-aRxh3ZmoBIEiVT-iCAvcqarLLNW1os,29756
204
+ ultralytics/utils/checks.py,sha256=-TLfWwHvAKqUb5m-78YTIrvsplp4VjDRtCNO2JsY3y0,29799
205
205
  ultralytics/utils/dist.py,sha256=NDFga-uKxkBX2zLxFHSene_cCiGQJoyOeCXcN9JIOIk,2358
206
206
  ultralytics/utils/downloads.py,sha256=fh7I5toTSowAOXtmx5zIzCEDREfTFG45cLIHmsDmuYw,21974
207
207
  ultralytics/utils/errors.py,sha256=GqP_Jgj_n0paxn8OMhn3DTCgoNkB2WjUcUaqs-M6SQk,816
@@ -226,10 +226,10 @@ ultralytics/utils/callbacks/mlflow.py,sha256=mkl_rK0Gy02cXnQUYmzmLE5W97fMgfEb7Il
226
226
  ultralytics/utils/callbacks/neptune.py,sha256=IbGQfEltamUKXJt93uSLQFn8c2rYh3DMTgVE1xsnmUI,3813
227
227
  ultralytics/utils/callbacks/raytune.py,sha256=ODVYzy-CoM4Uge0zjkh3Hnh9nF2M0vhDrSenXnvcizw,705
228
228
  ultralytics/utils/callbacks/tensorboard.py,sha256=SHlE58Fb-sg-uZKtgy-ybIO3SAIfK55aj8kTYGA0Cyg,4167
229
- ultralytics/utils/callbacks/wb.py,sha256=upfbF8-LLXueUvulLaMDmKDhKCl_PWbNa_87PQ0L0Rc,6752
230
- ultralytics-8.3.18.dist-info/LICENSE,sha256=DZak_2itbUtvHzD3E7GNUYSRK6jdOJ-GqncQ2weavLA,34523
231
- ultralytics-8.3.18.dist-info/METADATA,sha256=V0Akt0E2sH-fly-S-Mr1ywhM-L8ufd9n3G5bzTY9UDo,34818
232
- ultralytics-8.3.18.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
233
- ultralytics-8.3.18.dist-info/entry_points.txt,sha256=YM_wiKyTe9yRrsEfqvYolNO5ngwfoL4-NwgKzc8_7sI,93
234
- ultralytics-8.3.18.dist-info/top_level.txt,sha256=XP49TwiMw4QGsvTLSYiJhz1xF_k7ev5mQ8jJXaXi45Q,12
235
- ultralytics-8.3.18.dist-info/RECORD,,
229
+ ultralytics/utils/callbacks/wb.py,sha256=oX3JarCJGhzvW556XiEXQNaZblAaK_UETAt3kzkY61w,6869
230
+ ultralytics-8.3.20.dist-info/LICENSE,sha256=DZak_2itbUtvHzD3E7GNUYSRK6jdOJ-GqncQ2weavLA,34523
231
+ ultralytics-8.3.20.dist-info/METADATA,sha256=LkUQPNejvCdvWsdP-A44HVpIXlyKbHbvzEMEEyLScOk,34818
232
+ ultralytics-8.3.20.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
233
+ ultralytics-8.3.20.dist-info/entry_points.txt,sha256=YM_wiKyTe9yRrsEfqvYolNO5ngwfoL4-NwgKzc8_7sI,93
234
+ ultralytics-8.3.20.dist-info/top_level.txt,sha256=XP49TwiMw4QGsvTLSYiJhz1xF_k7ev5mQ8jJXaXi45Q,12
235
+ ultralytics-8.3.20.dist-info/RECORD,,