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 +1 -1
- ultralytics/engine/exporter.py +2 -2
- ultralytics/nn/autobackend.py +2 -2
- ultralytics/utils/callbacks/wb.py +13 -11
- ultralytics/utils/checks.py +5 -3
- {ultralytics-8.3.18.dist-info → ultralytics-8.3.20.dist-info}/METADATA +1 -1
- {ultralytics-8.3.18.dist-info → ultralytics-8.3.20.dist-info}/RECORD +11 -11
- {ultralytics-8.3.18.dist-info → ultralytics-8.3.20.dist-info}/LICENSE +0 -0
- {ultralytics-8.3.18.dist-info → ultralytics-8.3.20.dist-info}/WHEEL +0 -0
- {ultralytics-8.3.18.dist-info → ultralytics-8.3.20.dist-info}/entry_points.txt +0 -0
- {ultralytics-8.3.18.dist-info → ultralytics-8.3.20.dist-info}/top_level.txt +0 -0
ultralytics/__init__.py
CHANGED
ultralytics/engine/exporter.py
CHANGED
@@ -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
|
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__, "
|
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__}...")
|
ultralytics/nn/autobackend.py
CHANGED
@@ -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
|
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__, "
|
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
|
-
|
141
|
-
|
142
|
-
|
143
|
-
x,
|
144
|
-
|
145
|
-
|
146
|
-
|
147
|
-
|
148
|
-
|
149
|
-
|
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
|
|
ultralytics/utils/checks.py
CHANGED
@@ -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
|
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(
|
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.
|
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=
|
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=
|
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=
|
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
|
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=
|
230
|
-
ultralytics-8.3.
|
231
|
-
ultralytics-8.3.
|
232
|
-
ultralytics-8.3.
|
233
|
-
ultralytics-8.3.
|
234
|
-
ultralytics-8.3.
|
235
|
-
ultralytics-8.3.
|
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,,
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|