ultralytics 8.3.134__py3-none-any.whl → 8.3.135__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/cfg/__init__.py +1 -1
- ultralytics/engine/exporter.py +3 -3
- ultralytics/models/yolo/model.py +1 -1
- ultralytics/utils/checks.py +2 -0
- {ultralytics-8.3.134.dist-info → ultralytics-8.3.135.dist-info}/METADATA +1 -1
- {ultralytics-8.3.134.dist-info → ultralytics-8.3.135.dist-info}/RECORD +11 -11
- {ultralytics-8.3.134.dist-info → ultralytics-8.3.135.dist-info}/WHEEL +1 -1
- {ultralytics-8.3.134.dist-info → ultralytics-8.3.135.dist-info}/entry_points.txt +0 -0
- {ultralytics-8.3.134.dist-info → ultralytics-8.3.135.dist-info}/licenses/LICENSE +0 -0
- {ultralytics-8.3.134.dist-info → ultralytics-8.3.135.dist-info}/top_level.txt +0 -0
ultralytics/__init__.py
CHANGED
ultralytics/cfg/__init__.py
CHANGED
@@ -732,7 +732,7 @@ def handle_yolo_solutions(args: List[str]) -> None:
|
|
732
732
|
results = solution(frame, f_n := f_n + 1) if solution_name == "analytics" else solution(frame)
|
733
733
|
if solution_name != "crop":
|
734
734
|
vw.write(results.plot_im)
|
735
|
-
if cv2.waitKey(1) & 0xFF == ord("q"):
|
735
|
+
if solution.CFG["show"] and cv2.waitKey(1) & 0xFF == ord("q"):
|
736
736
|
break
|
737
737
|
finally:
|
738
738
|
cap.release()
|
ultralytics/engine/exporter.py
CHANGED
@@ -926,7 +926,7 @@ class Exporter:
|
|
926
926
|
"sng4onnx>=1.0.1", # required by 'onnx2tf' package
|
927
927
|
"onnx_graphsurgeon>=0.3.26", # required by 'onnx2tf' package
|
928
928
|
"ai-edge-litert>=1.2.0", # required by 'onnx2tf' package
|
929
|
-
"onnx>=1.12.0",
|
929
|
+
"onnx>=1.12.0,<1.18.0",
|
930
930
|
"onnx2tf>=1.26.3",
|
931
931
|
"onnxslim>=0.1.46",
|
932
932
|
"onnxruntime-gpu" if cuda else "onnxruntime",
|
@@ -943,8 +943,6 @@ class Exporter:
|
|
943
943
|
verbose=True,
|
944
944
|
msg="https://github.com/ultralytics/ultralytics/issues/5161",
|
945
945
|
)
|
946
|
-
import onnx2tf
|
947
|
-
|
948
946
|
f = Path(str(self.file).replace(self.file.suffix, "_saved_model"))
|
949
947
|
if f.is_dir():
|
950
948
|
shutil.rmtree(f) # delete output folder
|
@@ -971,6 +969,8 @@ class Exporter:
|
|
971
969
|
np.save(str(tmp_file), images.numpy().astype(np.float32)) # BHWC
|
972
970
|
np_data = [["images", tmp_file, [[[[0, 0, 0]]]], [[[[255, 255, 255]]]]]]
|
973
971
|
|
972
|
+
import onnx2tf # scoped for after ONNX export for reduced conflict during import
|
973
|
+
|
974
974
|
LOGGER.info(f"{prefix} starting TFLite export with onnx2tf {onnx2tf.__version__}...")
|
975
975
|
keras_model = onnx2tf.convert(
|
976
976
|
input_onnx_file_path=f_onnx,
|
ultralytics/models/yolo/model.py
CHANGED
@@ -331,7 +331,7 @@ class YOLOE(Model):
|
|
331
331
|
if len(visual_prompts):
|
332
332
|
num_cls = (
|
333
333
|
max(len(set(c)) for c in visual_prompts["cls"])
|
334
|
-
if isinstance(source, list) # means multiple images
|
334
|
+
if isinstance(source, list) and refer_image is None # means multiple images
|
335
335
|
else len(set(visual_prompts["cls"]))
|
336
336
|
)
|
337
337
|
self.model.model[-1].nc = num_cls
|
ultralytics/utils/checks.py
CHANGED
@@ -1,5 +1,6 @@
|
|
1
1
|
# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
|
2
2
|
|
3
|
+
import functools
|
3
4
|
import glob
|
4
5
|
import inspect
|
5
6
|
import math
|
@@ -296,6 +297,7 @@ def check_pip_update_available():
|
|
296
297
|
|
297
298
|
|
298
299
|
@ThreadingLocked()
|
300
|
+
@functools.lru_cache
|
299
301
|
def check_font(font="Arial.ttf"):
|
300
302
|
"""
|
301
303
|
Find font locally or download to user's configuration directory if it does not already exist.
|
@@ -1,6 +1,6 @@
|
|
1
1
|
Metadata-Version: 2.4
|
2
2
|
Name: ultralytics
|
3
|
-
Version: 8.3.
|
3
|
+
Version: 8.3.135
|
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,10 +7,10 @@ tests/test_exports.py,sha256=UeeBloqYYGZNh520R3CR80XBxA9XFrNmbK9An6V6C4w,9838
|
|
7
7
|
tests/test_integrations.py,sha256=dQteeRsRVuT_p5-T88-7jqT65Zm9iAXkyKg-KQ1_TQ8,6341
|
8
8
|
tests/test_python.py,sha256=KWsncKpeDdRmjRftmJpsMl7bBLI3TG_I7Lb4kuemZzQ,25618
|
9
9
|
tests/test_solutions.py,sha256=IFlqyOUCvGbLe_YZqWmNCe_afg4as0p-SfAv3j7VURI,6205
|
10
|
-
ultralytics/__init__.py,sha256=
|
10
|
+
ultralytics/__init__.py,sha256=7IMXy8Z7sekeQRLOVZyuYbA-1kse0gieArFyUxQ9dyE,730
|
11
11
|
ultralytics/assets/bus.jpg,sha256=wCAZxJecGR63Od3ZRERe9Aja1Weayrb9Ug751DS_vGM,137419
|
12
12
|
ultralytics/assets/zidane.jpg,sha256=Ftc4aeMmen1O0A3o6GCDO9FlfBslLpTAw0gnetx7bts,50427
|
13
|
-
ultralytics/cfg/__init__.py,sha256=
|
13
|
+
ultralytics/cfg/__init__.py,sha256=p1dKUDoVsnjJG8Qj5Q-eukb0WH2IoWV3BcJpEmKu2tE,39487
|
14
14
|
ultralytics/cfg/default.yaml,sha256=oFG6llJO-Py5H-cR9qs-7FieJamroDLwpbrkhmfROOM,8307
|
15
15
|
ultralytics/cfg/datasets/Argoverse.yaml,sha256=_xlEDIJ9XkUo0v_iNL7FW079BoSeZtKSuLteKTtGbA8,3275
|
16
16
|
ultralytics/cfg/datasets/DOTAv1.5.yaml,sha256=SHND_CFkojxw5iQD5Mcgju2kCZIl0gW2ajuzv1cqoL0,1224
|
@@ -118,7 +118,7 @@ ultralytics/data/scripts/get_coco.sh,sha256=UuJpJeo3qQpTHVINeOpmP0NYmg8PhEFE3A8J
|
|
118
118
|
ultralytics/data/scripts/get_coco128.sh,sha256=qmRQl_hOKrsdHrTrnyQuFIH01oDz3lfaz138OgGfLt8,650
|
119
119
|
ultralytics/data/scripts/get_imagenet.sh,sha256=hr42H16bM47iT27rgS7MpEo-GeOZAYUQXgr0B2cwn48,1705
|
120
120
|
ultralytics/engine/__init__.py,sha256=lm6MckFYCPTbqIoX7w0s_daxdjNeBeKW6DXppv1-QUM,70
|
121
|
-
ultralytics/engine/exporter.py,sha256=
|
121
|
+
ultralytics/engine/exporter.py,sha256=tXqZlcOZnDqtK7A0nwago7FfDAb3ftnYui-VeOFzVs0,70823
|
122
122
|
ultralytics/engine/model.py,sha256=fWhPNWUQzjjWfTEXzTaqSSearV4THRkEa_fl4dDvzWw,52930
|
123
123
|
ultralytics/engine/predictor.py,sha256=AwKpOGY2G-thNNiRw4Kf_MBLamq5tbRhXLNSMRArqFo,21803
|
124
124
|
ultralytics/engine/results.py,sha256=MhbyMCwgslmtV53fqii4UJUaLQ4gKTKdkXi7vvmJDAE,79628
|
@@ -163,7 +163,7 @@ ultralytics/models/utils/__init__.py,sha256=lm6MckFYCPTbqIoX7w0s_daxdjNeBeKW6DXp
|
|
163
163
|
ultralytics/models/utils/loss.py,sha256=FShJFvzFBk0HRepRhiSVNz9J-Cq08FxkSNXhLppycI0,19993
|
164
164
|
ultralytics/models/utils/ops.py,sha256=SuBnwwgUTqByNHpufobGLW72yO2cyfZFi14KAFWSjjw,13613
|
165
165
|
ultralytics/models/yolo/__init__.py,sha256=or0j5xvcM0usMlsFTYhNAOcQUri7reD0cD9JR5b7zDk,307
|
166
|
-
ultralytics/models/yolo/model.py,sha256=
|
166
|
+
ultralytics/models/yolo/model.py,sha256=oGPLPkgym0kfFhkLgyriR5KbKTQyJH-1Uggup5wFgw0,14296
|
167
167
|
ultralytics/models/yolo/classify/__init__.py,sha256=9--HVaNOfI1K7rn_rRqclL8FUAnpfeBrRqEQIaQw2xM,383
|
168
168
|
ultralytics/models/yolo/classify/predict.py,sha256=JV9szginTQ9Lpob0FozhKMiEIu1vVaYg4YItuVK2AFM,4081
|
169
169
|
ultralytics/models/yolo/classify/train.py,sha256=rv2CJv9fzvtHf2q4l5g0RsjplWKeLpz637kKqjtrLNY,9737
|
@@ -237,7 +237,7 @@ ultralytics/utils/__init__.py,sha256=vac0M-Hx55QXl6Vod3QPjnLBlt87Hwxu1784RXPmeQA
|
|
237
237
|
ultralytics/utils/autobatch.py,sha256=kg05q2qKg74y_Uq2vvr01i3KhLfpVR7sT0IXBt3_kyI,4921
|
238
238
|
ultralytics/utils/autodevice.py,sha256=OKZfTbswg6SlsYGCGMqROkA-451CXGG47oeyC5Q1kFM,7232
|
239
239
|
ultralytics/utils/benchmarks.py,sha256=lDNNnLeLUzmqKrqrqlCOiau-q7A-gcLooZP2dbxCu-U,30214
|
240
|
-
ultralytics/utils/checks.py,sha256=
|
240
|
+
ultralytics/utils/checks.py,sha256=L5G8CiQo8v2842KLGOaLG5y_AYRoa5gxCdtTt48LnS0,33129
|
241
241
|
ultralytics/utils/dist.py,sha256=aytW0JEkcA5ZTZucV92ot7Bn-apiej8aLk3QNWicjAc,4103
|
242
242
|
ultralytics/utils/downloads.py,sha256=Rn8xDwn2bzgBqiYz3Xn0rm3MWjk4T-QUd2Ajlu1EpQ4,22312
|
243
243
|
ultralytics/utils/errors.py,sha256=vY9h2evFSrHnZdHJVVrmm8Zzw4qVDLyo9DeYW5g0dFk,1573
|
@@ -264,9 +264,9 @@ ultralytics/utils/callbacks/neptune.py,sha256=yYUgEgSv6L39sSev6vjwhAWU3DlPDsbSDV
|
|
264
264
|
ultralytics/utils/callbacks/raytune.py,sha256=A8amUGpux7dYES-L1iSeMoMXBySGWCD1aUqT7vcG-pU,1284
|
265
265
|
ultralytics/utils/callbacks/tensorboard.py,sha256=jgYnym3cUQFAgN1GzTyO7l3jINtfAh8zhrllDvnLuVQ,5339
|
266
266
|
ultralytics/utils/callbacks/wb.py,sha256=iDRFXI4IIDm8R5OI89DMTmjs8aHLo1HRCLkOFKdaMG4,7507
|
267
|
-
ultralytics-8.3.
|
268
|
-
ultralytics-8.3.
|
269
|
-
ultralytics-8.3.
|
270
|
-
ultralytics-8.3.
|
271
|
-
ultralytics-8.3.
|
272
|
-
ultralytics-8.3.
|
267
|
+
ultralytics-8.3.135.dist-info/licenses/LICENSE,sha256=DZak_2itbUtvHzD3E7GNUYSRK6jdOJ-GqncQ2weavLA,34523
|
268
|
+
ultralytics-8.3.135.dist-info/METADATA,sha256=fukeo8lOpErZd4N34nFV-AH8hNzKpuYnOjWniPl0JQ4,37231
|
269
|
+
ultralytics-8.3.135.dist-info/WHEEL,sha256=Nw36Djuh_5VDukK0H78QzOX-_FQEo6V37m3nkm96gtU,91
|
270
|
+
ultralytics-8.3.135.dist-info/entry_points.txt,sha256=YM_wiKyTe9yRrsEfqvYolNO5ngwfoL4-NwgKzc8_7sI,93
|
271
|
+
ultralytics-8.3.135.dist-info/top_level.txt,sha256=XP49TwiMw4QGsvTLSYiJhz1xF_k7ev5mQ8jJXaXi45Q,12
|
272
|
+
ultralytics-8.3.135.dist-info/RECORD,,
|
File without changes
|
File without changes
|
File without changes
|