dgenerate-ultralytics-headless 8.3.248__py3-none-any.whl → 8.4.7__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.
- {dgenerate_ultralytics_headless-8.3.248.dist-info → dgenerate_ultralytics_headless-8.4.7.dist-info}/METADATA +52 -61
- {dgenerate_ultralytics_headless-8.3.248.dist-info → dgenerate_ultralytics_headless-8.4.7.dist-info}/RECORD +97 -84
- {dgenerate_ultralytics_headless-8.3.248.dist-info → dgenerate_ultralytics_headless-8.4.7.dist-info}/WHEEL +1 -1
- tests/__init__.py +2 -2
- tests/conftest.py +1 -1
- tests/test_cuda.py +8 -2
- tests/test_engine.py +8 -8
- tests/test_exports.py +11 -4
- tests/test_integrations.py +9 -9
- tests/test_python.py +41 -16
- tests/test_solutions.py +3 -3
- ultralytics/__init__.py +1 -1
- ultralytics/cfg/__init__.py +31 -31
- ultralytics/cfg/datasets/TT100K.yaml +346 -0
- ultralytics/cfg/datasets/coco12-formats.yaml +101 -0
- ultralytics/cfg/default.yaml +3 -1
- ultralytics/cfg/models/26/yolo26-cls.yaml +33 -0
- ultralytics/cfg/models/26/yolo26-obb.yaml +52 -0
- ultralytics/cfg/models/26/yolo26-p2.yaml +60 -0
- ultralytics/cfg/models/26/yolo26-p6.yaml +62 -0
- ultralytics/cfg/models/26/yolo26-pose.yaml +53 -0
- ultralytics/cfg/models/26/yolo26-seg.yaml +52 -0
- ultralytics/cfg/models/26/yolo26.yaml +52 -0
- ultralytics/cfg/models/26/yoloe-26-seg.yaml +53 -0
- ultralytics/cfg/models/26/yoloe-26.yaml +53 -0
- ultralytics/data/annotator.py +2 -2
- ultralytics/data/augment.py +15 -0
- ultralytics/data/converter.py +76 -45
- ultralytics/data/dataset.py +1 -1
- ultralytics/data/utils.py +2 -2
- ultralytics/engine/exporter.py +34 -28
- ultralytics/engine/model.py +38 -37
- ultralytics/engine/predictor.py +17 -17
- ultralytics/engine/results.py +22 -15
- ultralytics/engine/trainer.py +83 -48
- ultralytics/engine/tuner.py +20 -11
- ultralytics/engine/validator.py +16 -16
- ultralytics/models/fastsam/predict.py +1 -1
- ultralytics/models/yolo/classify/predict.py +1 -1
- ultralytics/models/yolo/classify/train.py +1 -1
- ultralytics/models/yolo/classify/val.py +1 -1
- ultralytics/models/yolo/detect/predict.py +2 -2
- ultralytics/models/yolo/detect/train.py +6 -3
- ultralytics/models/yolo/detect/val.py +7 -1
- ultralytics/models/yolo/model.py +8 -8
- ultralytics/models/yolo/obb/predict.py +2 -2
- ultralytics/models/yolo/obb/train.py +3 -3
- ultralytics/models/yolo/obb/val.py +1 -1
- ultralytics/models/yolo/pose/predict.py +1 -1
- ultralytics/models/yolo/pose/train.py +3 -1
- ultralytics/models/yolo/pose/val.py +1 -1
- ultralytics/models/yolo/segment/predict.py +3 -3
- ultralytics/models/yolo/segment/train.py +4 -4
- ultralytics/models/yolo/segment/val.py +2 -2
- ultralytics/models/yolo/yoloe/train.py +6 -1
- ultralytics/models/yolo/yoloe/train_seg.py +6 -1
- ultralytics/nn/autobackend.py +14 -8
- ultralytics/nn/modules/__init__.py +8 -0
- ultralytics/nn/modules/block.py +128 -8
- ultralytics/nn/modules/head.py +788 -203
- ultralytics/nn/tasks.py +86 -41
- ultralytics/nn/text_model.py +5 -2
- ultralytics/optim/__init__.py +5 -0
- ultralytics/optim/muon.py +338 -0
- ultralytics/solutions/ai_gym.py +3 -3
- ultralytics/solutions/config.py +1 -1
- ultralytics/solutions/heatmap.py +1 -1
- ultralytics/solutions/instance_segmentation.py +2 -2
- ultralytics/solutions/object_counter.py +1 -1
- ultralytics/solutions/parking_management.py +1 -1
- ultralytics/solutions/solutions.py +2 -2
- ultralytics/trackers/byte_tracker.py +7 -7
- ultralytics/trackers/track.py +1 -1
- ultralytics/utils/__init__.py +8 -8
- ultralytics/utils/benchmarks.py +26 -26
- ultralytics/utils/callbacks/platform.py +173 -64
- ultralytics/utils/callbacks/tensorboard.py +2 -0
- ultralytics/utils/callbacks/wb.py +6 -1
- ultralytics/utils/checks.py +28 -9
- ultralytics/utils/dist.py +1 -0
- ultralytics/utils/downloads.py +5 -3
- ultralytics/utils/export/engine.py +19 -10
- ultralytics/utils/export/imx.py +38 -20
- ultralytics/utils/export/tensorflow.py +21 -21
- ultralytics/utils/files.py +2 -2
- ultralytics/utils/loss.py +597 -203
- ultralytics/utils/metrics.py +2 -1
- ultralytics/utils/ops.py +11 -2
- ultralytics/utils/patches.py +42 -0
- ultralytics/utils/plotting.py +3 -0
- ultralytics/utils/tal.py +100 -20
- ultralytics/utils/torch_utils.py +1 -1
- ultralytics/utils/tqdm.py +4 -1
- ultralytics/utils/tuner.py +2 -5
- {dgenerate_ultralytics_headless-8.3.248.dist-info → dgenerate_ultralytics_headless-8.4.7.dist-info}/entry_points.txt +0 -0
- {dgenerate_ultralytics_headless-8.3.248.dist-info → dgenerate_ultralytics_headless-8.4.7.dist-info}/licenses/LICENSE +0 -0
- {dgenerate_ultralytics_headless-8.3.248.dist-info → dgenerate_ultralytics_headless-8.4.7.dist-info}/top_level.txt +0 -0
|
@@ -1,19 +1,19 @@
|
|
|
1
|
-
dgenerate_ultralytics_headless-8.
|
|
2
|
-
tests/__init__.py,sha256=
|
|
3
|
-
tests/conftest.py,sha256=
|
|
1
|
+
dgenerate_ultralytics_headless-8.4.7.dist-info/licenses/LICENSE,sha256=DZak_2itbUtvHzD3E7GNUYSRK6jdOJ-GqncQ2weavLA,34523
|
|
2
|
+
tests/__init__.py,sha256=hfUXxYLJB3846OCzWV94ZKEZsi8vq9Pqrdd2mMgjjck,804
|
|
3
|
+
tests/conftest.py,sha256=rlKyDuOC_3ptXrWS8Q19bNEGOupUmYXHj3nB6o1GBGY,2318
|
|
4
4
|
tests/test_cli.py,sha256=GhIFHi-_WIJpDgoGNRi0DnjbfwP1wHbklBMnkCM-P_4,5464
|
|
5
|
-
tests/test_cuda.py,sha256=
|
|
6
|
-
tests/test_engine.py,sha256=
|
|
7
|
-
tests/test_exports.py,sha256=
|
|
8
|
-
tests/test_integrations.py,sha256=
|
|
9
|
-
tests/test_python.py,sha256=
|
|
10
|
-
tests/test_solutions.py,sha256=
|
|
11
|
-
ultralytics/__init__.py,sha256=
|
|
5
|
+
tests/test_cuda.py,sha256=2TBe-ZkecMOGPWLdHcbsAjH3m9c5SQJ2KeyICgS0aeo,8426
|
|
6
|
+
tests/test_engine.py,sha256=ufSn3X4kL_Lpn2O25jKAfw_9QwHTMRjP9shDdpgBqnY,5740
|
|
7
|
+
tests/test_exports.py,sha256=Toy4u-4bsoyAbzNhc9kbMuKqvMKywZxNj5jlFNTzFWs,14670
|
|
8
|
+
tests/test_integrations.py,sha256=FjvTGjXm3bvYHK3_obgObhC5SzHCTzw4aOJV9Hh08jQ,6220
|
|
9
|
+
tests/test_python.py,sha256=CRgmOp2TiGBn9p7m16PVXBq3G9SkzIWG_kZvC9-nTGo,30474
|
|
10
|
+
tests/test_solutions.py,sha256=1tRlM72YciE42Nk9v83gsXOD5RSx9GSWVsKGhH7-HxE,14122
|
|
11
|
+
ultralytics/__init__.py,sha256=uc5Wwzw5ozOEVHw3LV11N34z-n4aZ3iH6pwcvQ4TP5I,1300
|
|
12
12
|
ultralytics/py.typed,sha256=la67KBlbjXN-_-DfGNcdOcjYumVpKG_Tkw-8n5dnGB4,8
|
|
13
13
|
ultralytics/assets/bus.jpg,sha256=wCAZxJecGR63Od3ZRERe9Aja1Weayrb9Ug751DS_vGM,137419
|
|
14
14
|
ultralytics/assets/zidane.jpg,sha256=Ftc4aeMmen1O0A3o6GCDO9FlfBslLpTAw0gnetx7bts,50427
|
|
15
|
-
ultralytics/cfg/__init__.py,sha256=
|
|
16
|
-
ultralytics/cfg/default.yaml,sha256=
|
|
15
|
+
ultralytics/cfg/__init__.py,sha256=N7eKXgd97UHWxYOgx_s3KKLzdKvRzp5LCFUL2P8Rpeo,40212
|
|
16
|
+
ultralytics/cfg/default.yaml,sha256=HFUxIYHNKR1RBmMfEmv72zNp2kqMzrSm18IQSKKTgnQ,9053
|
|
17
17
|
ultralytics/cfg/datasets/Argoverse.yaml,sha256=QGpdh3Hj5dFrvbsaE_8rAVj9BO4XpKTB7uhXaTTnE-o,3364
|
|
18
18
|
ultralytics/cfg/datasets/DOTAv1.5.yaml,sha256=KE7VC-ZMDSei1pLPm-pdk_ZAMRU_gLwGgtIQNbwp6dA,1212
|
|
19
19
|
ultralytics/cfg/datasets/DOTAv1.yaml,sha256=DUmBEfvdlCRH2t9aqhc3uk55sOXWWsY9v6RVYaELeTA,1182
|
|
@@ -22,6 +22,7 @@ ultralytics/cfg/datasets/HomeObjects-3K.yaml,sha256=xEtSqEad-rtfGuIrERjjhdISggmP
|
|
|
22
22
|
ultralytics/cfg/datasets/ImageNet.yaml,sha256=N9NHhIgnlNIBqZZbzQZAW3aCnz6RSXQABnopaDs5BmE,42529
|
|
23
23
|
ultralytics/cfg/datasets/Objects365.yaml,sha256=8Bl-NAm0mlMW8EfMsz39JZo-HCvmp0ejJXaMeoHTpqw,9649
|
|
24
24
|
ultralytics/cfg/datasets/SKU-110K.yaml,sha256=xvRkq3SdDOwBA91U85bln7HTXkod5MvFX6pt1PxTjJE,2609
|
|
25
|
+
ultralytics/cfg/datasets/TT100K.yaml,sha256=qrJ6nrZdvrMy5ov9FaHn-pFI8hJn_WLYaB60vhtCOxs,6918
|
|
25
26
|
ultralytics/cfg/datasets/VOC.yaml,sha256=XpaegRHjp7xZnenOuA9zgg2lQURSL-o7mLQwzIKKuqM,3803
|
|
26
27
|
ultralytics/cfg/datasets/VisDrone.yaml,sha256=PfudojW5av_5q-dC9VsG_xhvuv9cTGEpRp4loXCJ4Ng,3397
|
|
27
28
|
ultralytics/cfg/datasets/african-wildlife.yaml,sha256=6UfO_gnwJEDVq05p72IMJfkTIKZlXKNLSeKru-JyTrQ,915
|
|
@@ -29,6 +30,7 @@ ultralytics/cfg/datasets/brain-tumor.yaml,sha256=qrxPO_t9wxbn2kHFwP3vGTzSWj2ELTL
|
|
|
29
30
|
ultralytics/cfg/datasets/carparts-seg.yaml,sha256=A4e9hM1unTY2jjZIXGiKSarF6R-Ad9R99t57OgRJ37w,1253
|
|
30
31
|
ultralytics/cfg/datasets/coco-pose.yaml,sha256=rl1Pcnn8Hmst-Ian0-HvP6WQ2PKZxr1AjBEA406vwWw,1928
|
|
31
32
|
ultralytics/cfg/datasets/coco.yaml,sha256=woUMk6L3G3DMQDcThIKouZMcjTI5vP9XUdEVrzYGL50,2584
|
|
33
|
+
ultralytics/cfg/datasets/coco12-formats.yaml,sha256=Zd-41pX4PEUVIehyE4829QK_fUxiyZ79JVQSH-1UJVM,1953
|
|
32
34
|
ultralytics/cfg/datasets/coco128-seg.yaml,sha256=JsXu197vJX1YRuFvbEjsXyv4LUWIET-ruWZ9KqX6hYk,1986
|
|
33
35
|
ultralytics/cfg/datasets/coco128.yaml,sha256=ok_dzaBUzSd0DWfe531GT_uYTEoF5mIQcgoMHZyIVIA,1965
|
|
34
36
|
ultralytics/cfg/datasets/coco8-grayscale.yaml,sha256=8v6G6mOzZHQNdQM1YwdTBW_lsWWkLRnAimwZBHKtJg8,1961
|
|
@@ -63,6 +65,15 @@ ultralytics/cfg/models/12/yolo12-obb.yaml,sha256=JMviFAOmDbW0aMNzZNqispP0wxWw3mt
|
|
|
63
65
|
ultralytics/cfg/models/12/yolo12-pose.yaml,sha256=Mr9xjYclLQzxYhMqjIKQTdiTvtqZvEXBtclADFggaMA,2074
|
|
64
66
|
ultralytics/cfg/models/12/yolo12-seg.yaml,sha256=RBFFz4b95Dupfg0fmqCkZ4i1Zzai_QyJrI6Y2oLsocM,1984
|
|
65
67
|
ultralytics/cfg/models/12/yolo12.yaml,sha256=ZeA8LuymJXPNjZ5xkxkZHkcktDaKDzUBb2Kc3gCLC1w,1953
|
|
68
|
+
ultralytics/cfg/models/26/yolo26-cls.yaml,sha256=GmaLvnB62X6r9_mZwpx0b2LTRYIXzVYqSqR3Mcy336g,1432
|
|
69
|
+
ultralytics/cfg/models/26/yolo26-obb.yaml,sha256=3EKhFjHlOV0XBgb8J2B4iMd7dONiBC1iohYQStk6cyw,2147
|
|
70
|
+
ultralytics/cfg/models/26/yolo26-p2.yaml,sha256=Dbwvo9_t5e3NolZquMfqFQrQZuop155o1sTaK6tHYnk,2405
|
|
71
|
+
ultralytics/cfg/models/26/yolo26-p6.yaml,sha256=YFWaH6NAOHoLYmH4Te3edcj5-c7QJ-glzri-xPFk4U0,2457
|
|
72
|
+
ultralytics/cfg/models/26/yolo26-pose.yaml,sha256=ac9XyCyVrTUHkjEoSkYKtQrawZQ6Vr4hQZE1aO83b7s,2245
|
|
73
|
+
ultralytics/cfg/models/26/yolo26-seg.yaml,sha256=iL4euVjAg8dSq0oN-71-VDk0EqSruTGBjfCHqQYUqrg,2154
|
|
74
|
+
ultralytics/cfg/models/26/yolo26.yaml,sha256=sdH6DGns7WT5k5U2A4vY1pfDLnIgCzQYenZ95DGvfvA,2120
|
|
75
|
+
ultralytics/cfg/models/26/yoloe-26-seg.yaml,sha256=QlyK3awjLOjs9q0raj0we5QCG1tDy-NvQ2uQzKDoBYk,2223
|
|
76
|
+
ultralytics/cfg/models/26/yoloe-26.yaml,sha256=YWsBS4D3f7QhqLQzYT-LVdCOISo0VaTILIaLKmujDdo,2191
|
|
66
77
|
ultralytics/cfg/models/rt-detr/rtdetr-l.yaml,sha256=hAZti6u7lYIeYERsRrsdU9wekNFHURH_mq6Ow4XfhB4,2036
|
|
67
78
|
ultralytics/cfg/models/rt-detr/rtdetr-resnet101.yaml,sha256=Rtj3KCpxsvvFmYTJ2NKqoc0fk7-I5gaZiDsdgXFZ_6g,1689
|
|
68
79
|
ultralytics/cfg/models/rt-detr/rtdetr-resnet50.yaml,sha256=QLhmuMS9OEuLFbMuaDrjtzCizpYzddQcM6QyBL6rhPg,1685
|
|
@@ -108,28 +119,28 @@ ultralytics/cfg/models/v9/yolov9t.yaml,sha256=Q8GpSXE7fumhuJiQg4a2SkuS_UmnXqp-eo
|
|
|
108
119
|
ultralytics/cfg/trackers/botsort.yaml,sha256=tRxC-qT4Wz0mLn5x7ZEwrqgGKrmTDVY7gMge-mhpe7U,1431
|
|
109
120
|
ultralytics/cfg/trackers/bytetrack.yaml,sha256=7LS1ObP5u7BUFcmeY6L2m3bRuPUktnpJspFKd_ElVWc,908
|
|
110
121
|
ultralytics/data/__init__.py,sha256=ToR8zl0JhBHy42ZvV7zIwO_F3lbi5oNlGQNPK3dlddU,644
|
|
111
|
-
ultralytics/data/annotator.py,sha256=
|
|
112
|
-
ultralytics/data/augment.py,sha256=
|
|
122
|
+
ultralytics/data/annotator.py,sha256=iu1En-LzlR4RyR3ocftthnAog_peQHV9ForPRo_QcX8,2985
|
|
123
|
+
ultralytics/data/augment.py,sha256=XR52_BEmwFOrdMxEVRypm_kz6ROkTBgVped05R2xZWs,128566
|
|
113
124
|
ultralytics/data/base.py,sha256=pMs8yJOmAFPXdgfLCDtUemSvkPNDzxReP-fWzkNtonc,19723
|
|
114
125
|
ultralytics/data/build.py,sha256=s-tkSZPf3OfQyfXPXB9XxdW_gIcU6Xy_u21ekSgTnRo,17205
|
|
115
|
-
ultralytics/data/converter.py,sha256=
|
|
116
|
-
ultralytics/data/dataset.py,sha256=
|
|
126
|
+
ultralytics/data/converter.py,sha256=iO3wlF8-Z1wyEH4ueptzOXZd6vJttLOhu7XpWYtitL8,33886
|
|
127
|
+
ultralytics/data/dataset.py,sha256=r_BZy4FwMZ-dYkaJiz1E3jr2pI6dn7V3hZwf2RM9_RQ,36536
|
|
117
128
|
ultralytics/data/loaders.py,sha256=BQbhgjiLCGcRBPkGVG9Hr1jeNfG1nuZD3jstiWb7zS8,31889
|
|
118
129
|
ultralytics/data/split.py,sha256=HpR0ltf5oN1DpZstavFbBFC1YdpGPaATXxDOcAMwOqc,5101
|
|
119
130
|
ultralytics/data/split_dota.py,sha256=Qp9vGB2lzb5fQOrpNupKc8KN9ulqZoco9d4gRcx7JZk,12873
|
|
120
|
-
ultralytics/data/utils.py,sha256=
|
|
131
|
+
ultralytics/data/utils.py,sha256=QfypAt0fGCfb5PGw9o9Za-xnH5MUVTsETk-_ZBhtLko,36818
|
|
121
132
|
ultralytics/data/scripts/download_weights.sh,sha256=0y8XtZxOru7dVThXDFUXLHBuICgOIqZNUwpyL4Rh6lg,595
|
|
122
133
|
ultralytics/data/scripts/get_coco.sh,sha256=UuJpJeo3qQpTHVINeOpmP0NYmg8PhEFE3A8J3jKrnPw,1768
|
|
123
134
|
ultralytics/data/scripts/get_coco128.sh,sha256=qmRQl_hOKrsdHrTrnyQuFIH01oDz3lfaz138OgGfLt8,650
|
|
124
135
|
ultralytics/data/scripts/get_imagenet.sh,sha256=hr42H16bM47iT27rgS7MpEo-GeOZAYUQXgr0B2cwn48,1705
|
|
125
136
|
ultralytics/engine/__init__.py,sha256=lm6MckFYCPTbqIoX7w0s_daxdjNeBeKW6DXppv1-QUM,70
|
|
126
|
-
ultralytics/engine/exporter.py,sha256=
|
|
127
|
-
ultralytics/engine/model.py,sha256=
|
|
128
|
-
ultralytics/engine/predictor.py,sha256=
|
|
129
|
-
ultralytics/engine/results.py,sha256=
|
|
130
|
-
ultralytics/engine/trainer.py,sha256=
|
|
131
|
-
ultralytics/engine/tuner.py,sha256=
|
|
132
|
-
ultralytics/engine/validator.py,sha256=
|
|
137
|
+
ultralytics/engine/exporter.py,sha256=n_DtRhD0jT9sTFb8oQ_TYdQYTQJbsQzwqdISwR-mQY4,73330
|
|
138
|
+
ultralytics/engine/model.py,sha256=euDHUy7J5vVBvS_d-KbGZd_0BP5bF6Y3cTQ7VXtwZ4k,53210
|
|
139
|
+
ultralytics/engine/predictor.py,sha256=tXrHSTHJ-rDQ3lrPW9P5_ei_ewTwbY2sji6MExybJ28,22838
|
|
140
|
+
ultralytics/engine/results.py,sha256=Lg-Ke8TU6qaxu0wQtOH26unORj4FRYxd8RL0VxV74Zw,68333
|
|
141
|
+
ultralytics/engine/trainer.py,sha256=_pd1lvD2TWcE3v7P4OWqq-fPK5HLzeknxhSylpRuuNw,47309
|
|
142
|
+
ultralytics/engine/tuner.py,sha256=F4fyQaC5_GT74TULRO0VhzTv2S_a54cZDc3FjFoqaHE,21840
|
|
143
|
+
ultralytics/engine/validator.py,sha256=DiKsygbNJdRdwXoKoYOJA6bP_T7vMW3Syj_Qc_l7xTM,17761
|
|
133
144
|
ultralytics/hub/__init__.py,sha256=Z0K_E00jzQh90b18q3IDChwVmTvyIYp6C00sCV-n2F8,6709
|
|
134
145
|
ultralytics/hub/auth.py,sha256=ANzCeZA7lUzTWc_sFHbDuuyBh1jLl2sTpHkoUbIkFYE,6254
|
|
135
146
|
ultralytics/hub/session.py,sha256=OzBXAL9R135gRDdfNYUqyiSrxOyaiMFCVYSZua99sF0,18364
|
|
@@ -138,7 +149,7 @@ ultralytics/hub/google/__init__.py,sha256=r06Ld4TuZEBOqg4iagpeN-eMAkg43T2OTxOH4_
|
|
|
138
149
|
ultralytics/models/__init__.py,sha256=ljus_u1CIuP99k9fu6sCtzIeFZ-TCE28NZ8kefZHFNY,309
|
|
139
150
|
ultralytics/models/fastsam/__init__.py,sha256=Ku89Fy_X8ok3YPEUajjUZ5i4O08jdJMjJHt-3Z99Frk,231
|
|
140
151
|
ultralytics/models/fastsam/model.py,sha256=HN6CAHCTwMmyBCQlXx4wMBU7XqkvVHyUawRaxn2Gur8,3426
|
|
141
|
-
ultralytics/models/fastsam/predict.py,sha256=
|
|
152
|
+
ultralytics/models/fastsam/predict.py,sha256=zYhlXIrn69ryPnBEwEx4YkgYobPPE3_zvZAX2uAUIP4,8543
|
|
142
153
|
ultralytics/models/fastsam/utils.py,sha256=de9ieh4pBUuTNh5HTiNdRpWZhXAaSfNo3R1FNMt2GOE,879
|
|
143
154
|
ultralytics/models/fastsam/val.py,sha256=T76Yl4PtPezjGOcpXUxEobr0xnkR42Z-wnIz89cZ-IE,2028
|
|
144
155
|
ultralytics/models/nas/__init__.py,sha256=Q4ZQak8xNWtV5YSw_pFu0anbCyDxxEAuMMDfMzu6-0s,207
|
|
@@ -180,62 +191,64 @@ ultralytics/models/utils/__init__.py,sha256=lm6MckFYCPTbqIoX7w0s_daxdjNeBeKW6DXp
|
|
|
180
191
|
ultralytics/models/utils/loss.py,sha256=9CcqRXDj5-I-7eZuenInvyoLcPf22Ynf3rUFA5V22bI,21131
|
|
181
192
|
ultralytics/models/utils/ops.py,sha256=z-Ebjv_k14bWOoP6nszDzDBiy3yELcVtbj6M8PsRpvE,15207
|
|
182
193
|
ultralytics/models/yolo/__init__.py,sha256=YD407NDDiyjo0x_MR6usJaTpePKPgsfBUYehlCw7lRs,307
|
|
183
|
-
ultralytics/models/yolo/model.py,sha256
|
|
194
|
+
ultralytics/models/yolo/model.py,sha256=HXkglzJQqW1x7MJaKavI5aasA-0lSH21Xcv_dac3SFU,18504
|
|
184
195
|
ultralytics/models/yolo/classify/__init__.py,sha256=9--HVaNOfI1K7rn_rRqclL8FUAnpfeBrRqEQIaQw2xM,383
|
|
185
|
-
ultralytics/models/yolo/classify/predict.py,sha256=
|
|
186
|
-
ultralytics/models/yolo/classify/train.py,sha256=
|
|
187
|
-
ultralytics/models/yolo/classify/val.py,sha256=
|
|
196
|
+
ultralytics/models/yolo/classify/predict.py,sha256=HCStYkSqeg32SNTWfr4FDCkUMQ4wnKqceUK3T995us4,4137
|
|
197
|
+
ultralytics/models/yolo/classify/train.py,sha256=41ZxaIJkzkRxfgq6VffFX5Xfsrm9tNv3i3bdtUPAocE,8958
|
|
198
|
+
ultralytics/models/yolo/classify/val.py,sha256=akH2P3nff4oiZtV2toKB3Z9HIbsVcwsb1uvDwhamszw,10503
|
|
188
199
|
ultralytics/models/yolo/detect/__init__.py,sha256=GIRsLYR-kT4JJx7lh4ZZAFGBZj0aebokuU0A7JbjDVA,257
|
|
189
|
-
ultralytics/models/yolo/detect/predict.py,sha256=
|
|
190
|
-
ultralytics/models/yolo/detect/train.py,sha256
|
|
191
|
-
ultralytics/models/yolo/detect/val.py,sha256
|
|
200
|
+
ultralytics/models/yolo/detect/predict.py,sha256=2nxlMyw_zVKq1aeJFRTgb4EGL2vOFq4pLT9tArHBfF8,5385
|
|
201
|
+
ultralytics/models/yolo/detect/train.py,sha256=uz9PTsoLnIypxiOX2C7C7an3sarIUCQmiqmlZScE84c,10586
|
|
202
|
+
ultralytics/models/yolo/detect/val.py,sha256=54AOR6r3istE0pILJ1v4xzPdv7UcvtTEZ6E5OGj3Jgc,22818
|
|
192
203
|
ultralytics/models/yolo/obb/__init__.py,sha256=tQmpG8wVHsajWkZdmD6cjGohJ4ki64iSXQT8JY_dydo,221
|
|
193
|
-
ultralytics/models/yolo/obb/predict.py,sha256=
|
|
194
|
-
ultralytics/models/yolo/obb/train.py,sha256=
|
|
195
|
-
ultralytics/models/yolo/obb/val.py,sha256=
|
|
204
|
+
ultralytics/models/yolo/obb/predict.py,sha256=I7hWDr1zuy2WuwGom9uzXqomfr7qVMWb7iRl18xdTYw,2577
|
|
205
|
+
ultralytics/models/yolo/obb/train.py,sha256=HEDdPiP-yBbrUQWllcD1rc3gGrbzQmT6RBMTGtmVOu0,3452
|
|
206
|
+
ultralytics/models/yolo/obb/val.py,sha256=qYNe7ZcW3rhTLYPw15OeGfBaqaa_f1ADs4FF21h32e4,14513
|
|
196
207
|
ultralytics/models/yolo/pose/__init__.py,sha256=_9OFLj19XwvJHBRxQtVW5CV7rvJ_3hDPE97miit0sPc,227
|
|
197
|
-
ultralytics/models/yolo/pose/predict.py,sha256=
|
|
198
|
-
ultralytics/models/yolo/pose/train.py,sha256=
|
|
199
|
-
ultralytics/models/yolo/pose/val.py,sha256=
|
|
208
|
+
ultralytics/models/yolo/pose/predict.py,sha256=6EW9palcAoWX-gu5ROQvO6AxBSm719934hhqF-9OGjM,3118
|
|
209
|
+
ultralytics/models/yolo/pose/train.py,sha256=IlmsFlb0TsWZVy6PL3Trr_aXfwwGMBKAHyxnP7VPp_g,4747
|
|
210
|
+
ultralytics/models/yolo/pose/val.py,sha256=0luDccEPb_lUMjzaBb5VMsh9RdXVAbxb3Br57VKWNdc,12004
|
|
200
211
|
ultralytics/models/yolo/segment/__init__.py,sha256=3IThhZ1wlkY9FvmWm9cE-5-ZyE6F1FgzAtQ6jOOFzzw,275
|
|
201
|
-
ultralytics/models/yolo/segment/predict.py,sha256=
|
|
202
|
-
ultralytics/models/yolo/segment/train.py,sha256=
|
|
203
|
-
ultralytics/models/yolo/segment/val.py,sha256=
|
|
212
|
+
ultralytics/models/yolo/segment/predict.py,sha256=zLhmSTVEnaUumIX9SbjZH09kr2VrNdYWEss7FvseVuY,5428
|
|
213
|
+
ultralytics/models/yolo/segment/train.py,sha256=nS3qrT7Y3swCwjGZzeDQ2EunC9ilMsOiWs6LaTUCAE4,3021
|
|
214
|
+
ultralytics/models/yolo/segment/val.py,sha256=AvPS4rhV2PFpi0yixUfJhdczXctmZQSKgTjh7qVH0To,13204
|
|
204
215
|
ultralytics/models/yolo/world/__init__.py,sha256=nlh8I6t8hMGz_vZg8QSlsUW1R-2eKvn9CGUoPPQEGhA,131
|
|
205
216
|
ultralytics/models/yolo/world/train.py,sha256=80kswko6Zu7peXPBhXcfrTo5HO3Rg8C_cu4vPBQlk7M,7906
|
|
206
217
|
ultralytics/models/yolo/world/train_world.py,sha256=5Jj4gzEwDJtz37bEahL6Lf4xp-c1xiYjGKeg_w7Esns,8723
|
|
207
218
|
ultralytics/models/yolo/yoloe/__init__.py,sha256=zaZo1_ommaxNv7mD7xpdSomNF4s8mpOcCVTXspg0ncY,760
|
|
208
219
|
ultralytics/models/yolo/yoloe/predict.py,sha256=zeu_whH4e2SIWXV8MmJ1NNzoM_cNsiI2kOTjlAhV4qg,7065
|
|
209
|
-
ultralytics/models/yolo/yoloe/train.py,sha256=
|
|
210
|
-
ultralytics/models/yolo/yoloe/train_seg.py,sha256=
|
|
220
|
+
ultralytics/models/yolo/yoloe/train.py,sha256=99iSHQs--5VU_s82Q4w-fAJmyT5-y0TykTul8bo4xFo,13303
|
|
221
|
+
ultralytics/models/yolo/yoloe/train_seg.py,sha256=rV2Jnbuh6vvBMaupaZK_aRXBMevO0XhN2VUR43ZwlIY,5285
|
|
211
222
|
ultralytics/models/yolo/yoloe/val.py,sha256=utUFWeFKRFWZrPr1y3A8ztbTwdoWMYqzlwBN7CQ0tCA,9418
|
|
212
223
|
ultralytics/nn/__init__.py,sha256=538LZPUKKvc3JCMgiQ4VLGqRN2ZAaVLFcQbeNNHFkEA,545
|
|
213
|
-
ultralytics/nn/autobackend.py,sha256=
|
|
214
|
-
ultralytics/nn/tasks.py,sha256=
|
|
215
|
-
ultralytics/nn/text_model.py,sha256=
|
|
216
|
-
ultralytics/nn/modules/__init__.py,sha256=
|
|
224
|
+
ultralytics/nn/autobackend.py,sha256=MLS68iMNv6U0HyBK8nGjcyLOyImYIGEjP4398KqOkV0,45068
|
|
225
|
+
ultralytics/nn/tasks.py,sha256=PmlYScI7qTRCmYRR90Mw1QnqeRzvY0ojAMrgStBr11g,72010
|
|
226
|
+
ultralytics/nn/text_model.py,sha256=c--WzxjFEDb7p95u3YGcSsJLjj91zFNqXshij8Evrwg,15291
|
|
227
|
+
ultralytics/nn/modules/__init__.py,sha256=9KyQBxpomp5uJJ1PvMGuOFs2pR3NpqZcFHJlM6Q56c0,3322
|
|
217
228
|
ultralytics/nn/modules/activation.py,sha256=J6n-CJKFK0YbhwcRDqm9zEJM9pSAEycj5quQss_3x6E,2219
|
|
218
|
-
ultralytics/nn/modules/block.py,sha256=
|
|
229
|
+
ultralytics/nn/modules/block.py,sha256=9d1eelj3uRnf-HWTHYTjsBqLSpMCrwBQuX52MjeapN4,74499
|
|
219
230
|
ultralytics/nn/modules/conv.py,sha256=9WUlBzHD-wLgz0riLyttzASLIqBtXPK6Jk5EdyIiGCM,21100
|
|
220
|
-
ultralytics/nn/modules/head.py,sha256=
|
|
231
|
+
ultralytics/nn/modules/head.py,sha256=eJvXtr_ONGqQVdtsUpJtslplgVblti5sMxP9nkoSa0Y,78057
|
|
221
232
|
ultralytics/nn/modules/transformer.py,sha256=lAjTH-U8IkBp_1cXSOOFSus9tJf-s8WISKKcXPB84CM,31972
|
|
222
233
|
ultralytics/nn/modules/utils.py,sha256=EyhENse_RESlXjLHAJWvV07_tq1MVMmfzXgPR1fiT9w,6066
|
|
234
|
+
ultralytics/optim/__init__.py,sha256=Sl3Dx2eiaJd_u4VbmqcBqWWDF8FHnO5W0nBEL8_M_C4,130
|
|
235
|
+
ultralytics/optim/muon.py,sha256=Cuak4LOcVVEWIhYm4WzGmww7nhfR1N_uQOpLPX7gV-c,14243
|
|
223
236
|
ultralytics/solutions/__init__.py,sha256=Jj7OcRiYjHH-e104H4xTgjjR5W6aPB4mBRndbaSPmgU,1209
|
|
224
|
-
ultralytics/solutions/ai_gym.py,sha256=
|
|
237
|
+
ultralytics/solutions/ai_gym.py,sha256=fq9sIb0RBBvyd7SZShY8TO690lKbpPNOFap4OGi5CI8,5181
|
|
225
238
|
ultralytics/solutions/analytics.py,sha256=UaH-B6h8Ir9l00deRUeAIW6QQTIO_595HTp93sdwteM,12820
|
|
226
|
-
ultralytics/solutions/config.py,sha256=
|
|
239
|
+
ultralytics/solutions/config.py,sha256=wT_79zyoy_6diG5Iz9JZLzgCuGMaHj770lwRntVuNjQ,5396
|
|
227
240
|
ultralytics/solutions/distance_calculation.py,sha256=RcpRDodEHAJUug9tobtQKt5_bySNA8NMSRiaL347Q1U,5891
|
|
228
|
-
ultralytics/solutions/heatmap.py,sha256=
|
|
229
|
-
ultralytics/solutions/instance_segmentation.py,sha256=
|
|
241
|
+
ultralytics/solutions/heatmap.py,sha256=0f7v-0oAGj4no_h1Ll-BGsTmszSBoQ0tNa4azJYAQQw,5481
|
|
242
|
+
ultralytics/solutions/instance_segmentation.py,sha256=poxfCKl4gm7pHhjwULOeIPIRy9q_wOxqwtnUXXE9NhQ,3778
|
|
230
243
|
ultralytics/solutions/object_blurrer.py,sha256=EZrv3oU68kEaahAxlhk9cF5ZKFtoVaW8bDB4Css9xe0,3981
|
|
231
|
-
ultralytics/solutions/object_counter.py,sha256=
|
|
244
|
+
ultralytics/solutions/object_counter.py,sha256=OpMSLlenDK-cLvCgCOoKbqMXIZrngyqP8DP6ZeEnWL8,9355
|
|
232
245
|
ultralytics/solutions/object_cropper.py,sha256=WRbrfXAR5aD6PQBqJ-BvcVaiaqta_9YeTlXN2dY274s,3510
|
|
233
|
-
ultralytics/solutions/parking_management.py,sha256=
|
|
246
|
+
ultralytics/solutions/parking_management.py,sha256=Q0fEFKlv6dKKWuw_4jmWaeHQVXGppzuU7Vr_HqVYqHM,13770
|
|
234
247
|
ultralytics/solutions/queue_management.py,sha256=NlVX6PMEaffjoZjfQrVyayaDUdtc0JF8GzTQrZFjpCg,4371
|
|
235
248
|
ultralytics/solutions/region_counter.py,sha256=IAvlFwEYoNftDzfBbdo5MzLwcuidOHW9oTGyRCDzMRc,6025
|
|
236
249
|
ultralytics/solutions/security_alarm.py,sha256=QjUIVBWcy094VTcOkk_zOq3BmKKOeIaHpVi_QMWo_3Q,6293
|
|
237
250
|
ultralytics/solutions/similarity_search.py,sha256=Q2FOBUtEokegiJHlfDbPP0bKxr5F-sHN3-IvskDoe00,9644
|
|
238
|
-
ultralytics/solutions/solutions.py,sha256=
|
|
251
|
+
ultralytics/solutions/solutions.py,sha256=ktLwDhC0y4k2FbNd0sk7Y8GcEvBu9wL3rXyFGwlbnIQ,36984
|
|
239
252
|
ultralytics/solutions/speed_estimation.py,sha256=WrZECxKAq6P4QpeTbhkp3-Rqjnox7tdR25fUxzozlpU,5861
|
|
240
253
|
ultralytics/solutions/streamlit_inference.py,sha256=utJOe0Weu44_ABF9rDnAjwLjKyn3gwfaYaxFfFbx-9c,13060
|
|
241
254
|
ultralytics/solutions/trackzone.py,sha256=oqv-zZL99RVUMcN5ViAPmadzX6QNdAEozYrrg2pqO6k,3903
|
|
@@ -244,37 +257,37 @@ ultralytics/solutions/templates/similarity-search.html,sha256=mYuJI8H84cmu4kwPq2
|
|
|
244
257
|
ultralytics/trackers/__init__.py,sha256=n3BOO0TR-Sz5ANDYOkKDipM9nSHOePMEwqafbk-YEPs,255
|
|
245
258
|
ultralytics/trackers/basetrack.py,sha256=F-EW29F9E8GwXr5vzwLqW2rNwItu4KIx2MKce5pQXxI,4374
|
|
246
259
|
ultralytics/trackers/bot_sort.py,sha256=WImn-BOzGrK9dgMFfMPzKFE5awhXEB2VOi7AbOf_Cdc,11831
|
|
247
|
-
ultralytics/trackers/byte_tracker.py,sha256=
|
|
248
|
-
ultralytics/trackers/track.py,sha256=
|
|
260
|
+
ultralytics/trackers/byte_tracker.py,sha256=Ye8cwEPAitVpUsz4Yc8YXj9TEwyfk4zwitSKkIuKIu8,21061
|
|
261
|
+
ultralytics/trackers/track.py,sha256=xte5lkVBbOnrZ_tVLsHUmzvtNjbdksTVeSFQtLCLt_M,4742
|
|
249
262
|
ultralytics/trackers/utils/__init__.py,sha256=lm6MckFYCPTbqIoX7w0s_daxdjNeBeKW6DXppv1-QUM,70
|
|
250
263
|
ultralytics/trackers/utils/gmc.py,sha256=cvvhNXOhylVQti4pJQSNPx4yPqhhhw1k2yzY0JFl7Zo,13760
|
|
251
264
|
ultralytics/trackers/utils/kalman_filter.py,sha256=crgysL2bo0v1eTljOlP2YqIJDLBcHjl75MRpbxfaR_M,21514
|
|
252
265
|
ultralytics/trackers/utils/matching.py,sha256=x6uZOIx0O9oVmAcfY6tYMTJQE2cDTUlRR690Y5UkHLs,7129
|
|
253
|
-
ultralytics/utils/__init__.py,sha256=
|
|
266
|
+
ultralytics/utils/__init__.py,sha256=XLEK_pvptzNWhJaO8x0MWghREIyEDei0LOGnUnmU1Kg,55145
|
|
254
267
|
ultralytics/utils/autobatch.py,sha256=jiE4m_--H9UkXFDm_FqzcZk_hSTCGpS72XdVEKgZwAo,5114
|
|
255
268
|
ultralytics/utils/autodevice.py,sha256=rXlPuo-iX-vZ4BabmMGEGh9Uxpau4R7Zlt1KCo9Xfyc,8892
|
|
256
|
-
ultralytics/utils/benchmarks.py,sha256=
|
|
257
|
-
ultralytics/utils/checks.py,sha256=
|
|
269
|
+
ultralytics/utils/benchmarks.py,sha256=f4RykrjO1oEBxrTbH6qM_9vMxYKXO9F0ruFcM4xKF7A,32293
|
|
270
|
+
ultralytics/utils/checks.py,sha256=NWc0J-Nk4qHSVEXFDWfJkI7IjTNHFXajKjsSodDroBk,39411
|
|
258
271
|
ultralytics/utils/cpu.py,sha256=OksKOlX93AsbSsFuoYvLXRXgpkOibrZSwQyW6lipt4Q,3493
|
|
259
|
-
ultralytics/utils/dist.py,sha256=
|
|
260
|
-
ultralytics/utils/downloads.py,sha256=
|
|
272
|
+
ultralytics/utils/dist.py,sha256=sktf2a_uh-vLg6piQyiuRJ5JcMggFYmhS8Wepnb88WM,4220
|
|
273
|
+
ultralytics/utils/downloads.py,sha256=TWXkYwR5hEpVMWL6fbjdywDmZe02WhyL_8YuLVce-uM,23069
|
|
261
274
|
ultralytics/utils/errors.py,sha256=dUZcTWpbJJHqEuWHM6IbeoJJ4TzA_yHBP8E7tEEpBVs,1388
|
|
262
275
|
ultralytics/utils/events.py,sha256=6vqs_iSxoXIhQ804sOjApNZmXwNW9FUFtjaHPY8ta10,4665
|
|
263
|
-
ultralytics/utils/files.py,sha256=
|
|
276
|
+
ultralytics/utils/files.py,sha256=u7pjz13wgkLSBfe_beeZrzar32_gaJWoIVa3nvY3mh8,8190
|
|
264
277
|
ultralytics/utils/git.py,sha256=UdqeIiiEzg1qkerAZrg5YtTYPuJYwrpxW9N_6Pq6s8U,5501
|
|
265
278
|
ultralytics/utils/instance.py,sha256=11mhefvTI9ftMqSirXuiViAi0Fxlo6v84qvNxfRNUoE,18862
|
|
266
279
|
ultralytics/utils/logger.py,sha256=T5iaNnaqbCvx_FZf1dhVkr5FVxyxb4vO17t4SJfCIhg,19132
|
|
267
|
-
ultralytics/utils/loss.py,sha256=
|
|
268
|
-
ultralytics/utils/metrics.py,sha256=
|
|
280
|
+
ultralytics/utils/loss.py,sha256=Uh705dxpHPFLKecjsm_nCZ8JTYv0OHKNE9_ZZyMDiUo,57006
|
|
281
|
+
ultralytics/utils/metrics.py,sha256=puMGn1LfVIlDvx5K7US4RtK8HYW6cRl9OznfV0nUPvk,69261
|
|
269
282
|
ultralytics/utils/nms.py,sha256=zv1rOzMF6WU8Kdk41VzNf1H1EMt_vZHcbDFbg3mnN2o,14248
|
|
270
|
-
ultralytics/utils/ops.py,sha256=
|
|
271
|
-
ultralytics/utils/patches.py,sha256=
|
|
272
|
-
ultralytics/utils/plotting.py,sha256=
|
|
273
|
-
ultralytics/utils/tal.py,sha256=
|
|
274
|
-
ultralytics/utils/torch_utils.py,sha256=
|
|
275
|
-
ultralytics/utils/tqdm.py,sha256=
|
|
283
|
+
ultralytics/utils/ops.py,sha256=4xqb7kwrAWm8c_zxOWP5JoXozgsA1Slk2s4XFwmEZCs,26089
|
|
284
|
+
ultralytics/utils/patches.py,sha256=yXkznJNo3M74gvvzWmHoZYbWFu-KnO3KK4usbmey8H0,8521
|
|
285
|
+
ultralytics/utils/plotting.py,sha256=_iXs4gs8tzMSgiKxCriD4un-MJkOsC3lGSy0wn7qZGk,48433
|
|
286
|
+
ultralytics/utils/tal.py,sha256=vfcfSy78zdtHbGzlvo5UDx-sCwHLRdGBqDO3CX7ZiR0,24182
|
|
287
|
+
ultralytics/utils/torch_utils.py,sha256=dHvLaQopIOr9NcIWkLWPX36f5OAFR4thcqm379Zayfc,40278
|
|
288
|
+
ultralytics/utils/tqdm.py,sha256=f2W608Qpvgu6tFi28qylaZpcRv3IX8wTGY_8lgicaqY,16343
|
|
276
289
|
ultralytics/utils/triton.py,sha256=BQu3CD3OlT76d1OtmnX5slQU37VC1kzRvEtfI2saIQA,5211
|
|
277
|
-
ultralytics/utils/tuner.py,sha256=
|
|
290
|
+
ultralytics/utils/tuner.py,sha256=nRMmnyp0B0gVJzAXcpCxQUnwXjVp0WNiSJwxyR2xvQM,7303
|
|
278
291
|
ultralytics/utils/callbacks/__init__.py,sha256=hzL63Rce6VkZhP4Lcim9LKjadixaQG86nKqPhk7IkS0,242
|
|
279
292
|
ultralytics/utils/callbacks/base.py,sha256=floD31JHqHpiVabQiE76_hzC_j7KjtL4w_czkD1bLKc,6883
|
|
280
293
|
ultralytics/utils/callbacks/clearml.py,sha256=LjfNe4mswceCOpEGVLxqGXjkl_XGbef4awdcp4502RU,5831
|
|
@@ -283,16 +296,16 @@ ultralytics/utils/callbacks/dvc.py,sha256=YT0Sa5P8Huj8Fn9jM2P6MYzUY3PIVxsa5BInVi
|
|
|
283
296
|
ultralytics/utils/callbacks/hub.py,sha256=fVLqqr3ZM6hoYFlVMEeejfq1MWDrkWCskPFOG3HGILQ,4159
|
|
284
297
|
ultralytics/utils/callbacks/mlflow.py,sha256=wCXjQgdufp9LYujqMzLZOmIOur6kvrApHNeo9dA7t_g,5323
|
|
285
298
|
ultralytics/utils/callbacks/neptune.py,sha256=_vt3cMwDHCR-LyT3KtRikGpj6AG11oQ-skUUUUdZ74o,4391
|
|
286
|
-
ultralytics/utils/callbacks/platform.py,sha256=
|
|
299
|
+
ultralytics/utils/callbacks/platform.py,sha256=Utc9X3SDEGcvyQLaujQs3IA8UpFvmJcQC6HmLnTV4XA,16202
|
|
287
300
|
ultralytics/utils/callbacks/raytune.py,sha256=Y0dFyNZVRuFovSh7nkgUIHTQL3xIXOACElgHuYbg_5I,1278
|
|
288
|
-
ultralytics/utils/callbacks/tensorboard.py,sha256=
|
|
289
|
-
ultralytics/utils/callbacks/wb.py,sha256=
|
|
301
|
+
ultralytics/utils/callbacks/tensorboard.py,sha256=K7b6KtC7rimfzqFu-NDZ_55Tbd7eC6TckqQdTNPuQ6U,5039
|
|
302
|
+
ultralytics/utils/callbacks/wb.py,sha256=ci6lYVRneKTRC5CL6FRf9_iOYznwU74p9_fV3s9AbfQ,7907
|
|
290
303
|
ultralytics/utils/export/__init__.py,sha256=Cfh-PwVfTF_lwPp-Ss4wiX4z8Sm1XRPklsqdFfmTZ30,333
|
|
291
|
-
ultralytics/utils/export/engine.py,sha256=
|
|
292
|
-
ultralytics/utils/export/imx.py,sha256=
|
|
293
|
-
ultralytics/utils/export/tensorflow.py,sha256=
|
|
294
|
-
dgenerate_ultralytics_headless-8.
|
|
295
|
-
dgenerate_ultralytics_headless-8.
|
|
296
|
-
dgenerate_ultralytics_headless-8.
|
|
297
|
-
dgenerate_ultralytics_headless-8.
|
|
298
|
-
dgenerate_ultralytics_headless-8.
|
|
304
|
+
ultralytics/utils/export/engine.py,sha256=QoXPqnmQn6W5TOUAygOtCG63R9ExDG4-Df6X6W-_Mzo,10470
|
|
305
|
+
ultralytics/utils/export/imx.py,sha256=VnMDO7c8ezBs91UDoLg9rR0oY8Uc7FujKpbdGxrzV18,13744
|
|
306
|
+
ultralytics/utils/export/tensorflow.py,sha256=xHEcEM3_VeYctyqkJCpgkqcNie1M8xLqcFKr6uANEEQ,9951
|
|
307
|
+
dgenerate_ultralytics_headless-8.4.7.dist-info/METADATA,sha256=-XtPXXF8jhWfqBElNXs7CDCIEsTvslaRYwzQXbwErKU,40081
|
|
308
|
+
dgenerate_ultralytics_headless-8.4.7.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
|
|
309
|
+
dgenerate_ultralytics_headless-8.4.7.dist-info/entry_points.txt,sha256=YM_wiKyTe9yRrsEfqvYolNO5ngwfoL4-NwgKzc8_7sI,93
|
|
310
|
+
dgenerate_ultralytics_headless-8.4.7.dist-info/top_level.txt,sha256=XP49TwiMw4QGsvTLSYiJhz1xF_k7ev5mQ8jJXaXi45Q,12
|
|
311
|
+
dgenerate_ultralytics_headless-8.4.7.dist-info/RECORD,,
|
tests/__init__.py
CHANGED
|
@@ -4,8 +4,8 @@ from ultralytics.cfg import TASK2DATA, TASK2MODEL, TASKS
|
|
|
4
4
|
from ultralytics.utils import ASSETS, WEIGHTS_DIR, checks
|
|
5
5
|
|
|
6
6
|
# Constants used in tests
|
|
7
|
-
MODEL = WEIGHTS_DIR / "path with spaces" / "
|
|
8
|
-
CFG = "
|
|
7
|
+
MODEL = WEIGHTS_DIR / "path with spaces" / "yolo26n.pt" # test spaces in path
|
|
8
|
+
CFG = "yolo26n.yaml"
|
|
9
9
|
SOURCE = ASSETS / "bus.jpg"
|
|
10
10
|
SOURCES_LIST = [ASSETS / "bus.jpg", ASSETS, ASSETS / "*", ASSETS / "**/*.jpg"]
|
|
11
11
|
CUDA_IS_AVAILABLE = checks.cuda_is_available()
|
tests/conftest.py
CHANGED
|
@@ -50,7 +50,7 @@ def pytest_terminal_summary(terminalreporter, exitstatus, config):
|
|
|
50
50
|
|
|
51
51
|
# Remove files
|
|
52
52
|
models = [path for x in {"*.onnx", "*.torchscript"} for path in WEIGHTS_DIR.rglob(x)]
|
|
53
|
-
for file in ["decelera_portrait_min.mov", "bus.jpg", "
|
|
53
|
+
for file in ["decelera_portrait_min.mov", "bus.jpg", "yolo26n.onnx", "yolo26n.torchscript", *models]:
|
|
54
54
|
Path(file).unlink(missing_ok=True)
|
|
55
55
|
|
|
56
56
|
# Remove directories
|
tests/test_cuda.py
CHANGED
|
@@ -41,7 +41,7 @@ def test_checks():
|
|
|
41
41
|
@pytest.mark.skipif(not DEVICES, reason="No CUDA devices available")
|
|
42
42
|
def test_amp():
|
|
43
43
|
"""Test AMP training checks."""
|
|
44
|
-
model = YOLO("
|
|
44
|
+
model = YOLO("yolo26n.pt").model.to(f"cuda:{DEVICES[0]}")
|
|
45
45
|
assert check_amp(model)
|
|
46
46
|
|
|
47
47
|
|
|
@@ -91,6 +91,12 @@ def test_export_onnx_matrix(task, dynamic, int8, half, batch, simplify, nms):
|
|
|
91
91
|
)
|
|
92
92
|
def test_export_engine_matrix(task, dynamic, int8, half, batch):
|
|
93
93
|
"""Test YOLO model export to TensorRT format for various configurations and run inference."""
|
|
94
|
+
import tensorrt as trt
|
|
95
|
+
|
|
96
|
+
is_trt10 = int(trt.__version__.split(".", 1)[0]) >= 10
|
|
97
|
+
if is_trt10 and int8 and dynamic:
|
|
98
|
+
pytest.skip("YOLO26 INT8+dynamic export requires explicit quantization on TensorRT 10+")
|
|
99
|
+
|
|
94
100
|
file = YOLO(TASK2MODEL[task]).export(
|
|
95
101
|
format="engine",
|
|
96
102
|
imgsz=32,
|
|
@@ -126,7 +132,7 @@ def test_train():
|
|
|
126
132
|
@pytest.mark.skipif(not DEVICES, reason="No CUDA devices available")
|
|
127
133
|
def test_predict_multiple_devices():
|
|
128
134
|
"""Validate model prediction consistency across CPU and CUDA devices."""
|
|
129
|
-
model = YOLO("
|
|
135
|
+
model = YOLO("yolo26n.pt")
|
|
130
136
|
|
|
131
137
|
# Test CPU
|
|
132
138
|
model = model.cpu()
|
tests/test_engine.py
CHANGED
|
@@ -5,7 +5,7 @@ from unittest import mock
|
|
|
5
5
|
|
|
6
6
|
import torch
|
|
7
7
|
|
|
8
|
-
from tests import MODEL
|
|
8
|
+
from tests import MODEL, SOURCE
|
|
9
9
|
from ultralytics import YOLO
|
|
10
10
|
from ultralytics.cfg import get_cfg
|
|
11
11
|
from ultralytics.engine.exporter import Exporter
|
|
@@ -23,13 +23,13 @@ def test_export():
|
|
|
23
23
|
exporter = Exporter()
|
|
24
24
|
exporter.add_callback("on_export_start", test_func)
|
|
25
25
|
assert test_func in exporter.callbacks["on_export_start"], "callback test failed"
|
|
26
|
-
f = exporter(model=YOLO("
|
|
27
|
-
YOLO(f)(
|
|
26
|
+
f = exporter(model=YOLO("yolo26n.yaml").model)
|
|
27
|
+
YOLO(f)(SOURCE) # exported model inference
|
|
28
28
|
|
|
29
29
|
|
|
30
30
|
def test_detect():
|
|
31
31
|
"""Test YOLO object detection training, validation, and prediction functionality."""
|
|
32
|
-
overrides = {"data": "coco8.yaml", "model": "
|
|
32
|
+
overrides = {"data": "coco8.yaml", "model": "yolo26n.yaml", "imgsz": 32, "epochs": 1, "save": False}
|
|
33
33
|
cfg = get_cfg(DEFAULT_CFG)
|
|
34
34
|
cfg.data = "coco8.yaml"
|
|
35
35
|
cfg.imgsz = 32
|
|
@@ -71,7 +71,7 @@ def test_segment():
|
|
|
71
71
|
"""Test image segmentation training, validation, and prediction pipelines using YOLO models."""
|
|
72
72
|
overrides = {
|
|
73
73
|
"data": "coco8-seg.yaml",
|
|
74
|
-
"model": "
|
|
74
|
+
"model": "yolo26n-seg.yaml",
|
|
75
75
|
"imgsz": 32,
|
|
76
76
|
"epochs": 1,
|
|
77
77
|
"save": False,
|
|
@@ -98,7 +98,7 @@ def test_segment():
|
|
|
98
98
|
pred = segment.SegmentationPredictor(overrides={"imgsz": [64, 64]})
|
|
99
99
|
pred.add_callback("on_predict_start", test_func)
|
|
100
100
|
assert test_func in pred.callbacks["on_predict_start"], "callback test failed"
|
|
101
|
-
result = pred(source=ASSETS, model=WEIGHTS_DIR / "
|
|
101
|
+
result = pred(source=ASSETS, model=WEIGHTS_DIR / "yolo26n-seg.pt")
|
|
102
102
|
assert len(result), "predictor test failed"
|
|
103
103
|
|
|
104
104
|
# Test resume functionality
|
|
@@ -115,7 +115,7 @@ def test_segment():
|
|
|
115
115
|
|
|
116
116
|
def test_classify():
|
|
117
117
|
"""Test image classification including training, validation, and prediction phases."""
|
|
118
|
-
overrides = {"data": "imagenet10", "model": "
|
|
118
|
+
overrides = {"data": "imagenet10", "model": "yolo26n-cls.yaml", "imgsz": 32, "epochs": 1, "save": False}
|
|
119
119
|
cfg = get_cfg(DEFAULT_CFG)
|
|
120
120
|
cfg.data = "imagenet10"
|
|
121
121
|
cfg.imgsz = 32
|
|
@@ -150,7 +150,7 @@ def test_nan_recovery():
|
|
|
150
150
|
trainer.tloss *= torch.tensor(float("nan"))
|
|
151
151
|
nan_injected[0] = True
|
|
152
152
|
|
|
153
|
-
overrides = {"data": "coco8.yaml", "model": "
|
|
153
|
+
overrides = {"data": "coco8.yaml", "model": "yolo26n.yaml", "imgsz": 32, "epochs": 3}
|
|
154
154
|
trainer = detect.DetectionTrainer(overrides=overrides)
|
|
155
155
|
trainer.add_callback("on_train_batch_end", inject_nan)
|
|
156
156
|
trainer.train()
|
tests/test_exports.py
CHANGED
|
@@ -12,8 +12,8 @@ import pytest
|
|
|
12
12
|
from tests import MODEL, SOURCE
|
|
13
13
|
from ultralytics import YOLO
|
|
14
14
|
from ultralytics.cfg import TASK2DATA, TASK2MODEL, TASKS
|
|
15
|
-
from ultralytics.utils import ARM64, IS_RASPBERRYPI, LINUX, MACOS, WINDOWS, checks
|
|
16
|
-
from ultralytics.utils.torch_utils import TORCH_1_10, TORCH_1_11, TORCH_1_13, TORCH_2_1, TORCH_2_8, TORCH_2_9
|
|
15
|
+
from ultralytics.utils import ARM64, IS_RASPBERRYPI, LINUX, MACOS, MACOS_VERSION, WINDOWS, checks
|
|
16
|
+
from ultralytics.utils.torch_utils import TORCH_1_10, TORCH_1_11, TORCH_1_13, TORCH_2_0, TORCH_2_1, TORCH_2_8, TORCH_2_9
|
|
17
17
|
|
|
18
18
|
|
|
19
19
|
def test_export_torchscript():
|
|
@@ -112,6 +112,9 @@ def test_export_torchscript_matrix(task, dynamic, int8, half, batch, nms):
|
|
|
112
112
|
@pytest.mark.skipif(not MACOS, reason="CoreML inference only supported on macOS")
|
|
113
113
|
@pytest.mark.skipif(not TORCH_1_11, reason="CoreML export requires torch>=1.11")
|
|
114
114
|
@pytest.mark.skipif(checks.IS_PYTHON_3_13, reason="CoreML not supported in Python 3.13")
|
|
115
|
+
@pytest.mark.skipif(
|
|
116
|
+
MACOS and MACOS_VERSION and MACOS_VERSION >= "15", reason="CoreML YOLO26 matrix test crashes on macOS 15+"
|
|
117
|
+
)
|
|
115
118
|
@pytest.mark.parametrize(
|
|
116
119
|
"task, dynamic, int8, half, nms, batch",
|
|
117
120
|
[ # generate all combinations except for exclusion cases
|
|
@@ -141,7 +144,9 @@ def test_export_coreml_matrix(task, dynamic, int8, half, nms, batch):
|
|
|
141
144
|
|
|
142
145
|
|
|
143
146
|
@pytest.mark.slow
|
|
144
|
-
@pytest.mark.skipif(
|
|
147
|
+
@pytest.mark.skipif(
|
|
148
|
+
not checks.IS_PYTHON_MINIMUM_3_10 or not TORCH_1_13, reason="TFLite export requires Python>=3.10 and torch>=1.13"
|
|
149
|
+
)
|
|
145
150
|
@pytest.mark.skipif(
|
|
146
151
|
not LINUX or IS_RASPBERRYPI,
|
|
147
152
|
reason="Test disabled as TF suffers from install conflicts on Windows, macOS and Raspberry Pi",
|
|
@@ -235,6 +240,7 @@ def test_export_mnn_matrix(task, int8, half, batch):
|
|
|
235
240
|
|
|
236
241
|
|
|
237
242
|
@pytest.mark.slow
|
|
243
|
+
@pytest.mark.skipif(not TORCH_2_0, reason="NCNN inference causes segfault on PyTorch<2.0")
|
|
238
244
|
def test_export_ncnn():
|
|
239
245
|
"""Test YOLO export to NCNN format."""
|
|
240
246
|
file = YOLO(MODEL).export(format="ncnn", imgsz=32)
|
|
@@ -242,6 +248,7 @@ def test_export_ncnn():
|
|
|
242
248
|
|
|
243
249
|
|
|
244
250
|
@pytest.mark.slow
|
|
251
|
+
@pytest.mark.skipif(not TORCH_2_0, reason="NCNN inference causes segfault on PyTorch<2.0")
|
|
245
252
|
@pytest.mark.parametrize("task, half, batch", list(product(TASKS, [True, False], [1])))
|
|
246
253
|
def test_export_ncnn_matrix(task, half, batch):
|
|
247
254
|
"""Test YOLO export to NCNN format considering various export configurations."""
|
|
@@ -256,7 +263,7 @@ def test_export_ncnn_matrix(task, half, batch):
|
|
|
256
263
|
@pytest.mark.skipif(ARM64, reason="IMX export is not supported on ARM64 architectures.")
|
|
257
264
|
def test_export_imx():
|
|
258
265
|
"""Test YOLO export to IMX format."""
|
|
259
|
-
model = YOLO(
|
|
266
|
+
model = YOLO("yolo11n.pt") # IMX export only supports YOLO11
|
|
260
267
|
file = model.export(format="imx", imgsz=32)
|
|
261
268
|
YOLO(file)(SOURCE, imgsz=32)
|
|
262
269
|
|
tests/test_integrations.py
CHANGED
|
@@ -18,14 +18,14 @@ from ultralytics.utils.checks import check_requirements
|
|
|
18
18
|
def test_tensorboard():
|
|
19
19
|
"""Test training with TensorBoard logging enabled."""
|
|
20
20
|
SETTINGS["tensorboard"] = True
|
|
21
|
-
YOLO("
|
|
21
|
+
YOLO("yolo26n-cls.yaml").train(data="imagenet10", imgsz=32, epochs=3, plots=False, device="cpu")
|
|
22
22
|
SETTINGS["tensorboard"] = False
|
|
23
23
|
|
|
24
24
|
|
|
25
25
|
@pytest.mark.skipif(not check_requirements("ray", install=False), reason="ray[tune] not installed")
|
|
26
26
|
def test_model_ray_tune():
|
|
27
27
|
"""Tune YOLO model using Ray for hyperparameter optimization."""
|
|
28
|
-
YOLO("
|
|
28
|
+
YOLO("yolo26n-cls.yaml").tune(
|
|
29
29
|
use_ray=True, data="imagenet10", grace_period=1, iterations=1, imgsz=32, epochs=1, plots=False, device="cpu"
|
|
30
30
|
)
|
|
31
31
|
|
|
@@ -34,7 +34,7 @@ def test_model_ray_tune():
|
|
|
34
34
|
def test_mlflow():
|
|
35
35
|
"""Test training with MLflow tracking enabled."""
|
|
36
36
|
SETTINGS["mlflow"] = True
|
|
37
|
-
YOLO("
|
|
37
|
+
YOLO("yolo26n-cls.yaml").train(data="imagenet10", imgsz=32, epochs=3, plots=False, device="cpu")
|
|
38
38
|
SETTINGS["mlflow"] = False
|
|
39
39
|
|
|
40
40
|
|
|
@@ -50,7 +50,7 @@ def test_mlflow_keep_run_active():
|
|
|
50
50
|
|
|
51
51
|
# Test with MLFLOW_KEEP_RUN_ACTIVE=True
|
|
52
52
|
os.environ["MLFLOW_KEEP_RUN_ACTIVE"] = "True"
|
|
53
|
-
YOLO("
|
|
53
|
+
YOLO("yolo26n-cls.yaml").train(data="imagenet10", imgsz=32, epochs=1, plots=False, device="cpu")
|
|
54
54
|
status = mlflow.active_run().info.status
|
|
55
55
|
assert status == "RUNNING", "MLflow run should be active when MLFLOW_KEEP_RUN_ACTIVE=True"
|
|
56
56
|
|
|
@@ -58,13 +58,13 @@ def test_mlflow_keep_run_active():
|
|
|
58
58
|
|
|
59
59
|
# Test with MLFLOW_KEEP_RUN_ACTIVE=False
|
|
60
60
|
os.environ["MLFLOW_KEEP_RUN_ACTIVE"] = "False"
|
|
61
|
-
YOLO("
|
|
61
|
+
YOLO("yolo26n-cls.yaml").train(data="imagenet10", imgsz=32, epochs=1, plots=False, device="cpu")
|
|
62
62
|
status = mlflow.get_run(run_id=run_id).info.status
|
|
63
63
|
assert status == "FINISHED", "MLflow run should be ended when MLFLOW_KEEP_RUN_ACTIVE=False"
|
|
64
64
|
|
|
65
65
|
# Test with MLFLOW_KEEP_RUN_ACTIVE not set
|
|
66
66
|
os.environ.pop("MLFLOW_KEEP_RUN_ACTIVE", None)
|
|
67
|
-
YOLO("
|
|
67
|
+
YOLO("yolo26n-cls.yaml").train(data="imagenet10", imgsz=32, epochs=1, plots=False, device="cpu")
|
|
68
68
|
status = mlflow.get_run(run_id=run_id).info.status
|
|
69
69
|
assert status == "FINISHED", "MLflow run should be ended by default when MLFLOW_KEEP_RUN_ACTIVE is not set"
|
|
70
70
|
SETTINGS["mlflow"] = False
|
|
@@ -129,21 +129,21 @@ def test_faster_coco_eval():
|
|
|
129
129
|
from ultralytics.models.yolo.pose import PoseValidator
|
|
130
130
|
from ultralytics.models.yolo.segment import SegmentationValidator
|
|
131
131
|
|
|
132
|
-
args = {"model": "
|
|
132
|
+
args = {"model": "yolo26n.pt", "data": "coco8.yaml", "save_json": True, "imgsz": 64}
|
|
133
133
|
validator = DetectionValidator(args=args)
|
|
134
134
|
validator()
|
|
135
135
|
validator.is_coco = True
|
|
136
136
|
download(f"{ASSETS_URL}/instances_val2017.json", dir=DATASETS_DIR / "coco8/annotations")
|
|
137
137
|
_ = validator.eval_json(validator.stats)
|
|
138
138
|
|
|
139
|
-
args = {"model": "
|
|
139
|
+
args = {"model": "yolo26n-seg.pt", "data": "coco8-seg.yaml", "save_json": True, "imgsz": 64}
|
|
140
140
|
validator = SegmentationValidator(args=args)
|
|
141
141
|
validator()
|
|
142
142
|
validator.is_coco = True
|
|
143
143
|
download(f"{ASSETS_URL}/instances_val2017.json", dir=DATASETS_DIR / "coco8-seg/annotations")
|
|
144
144
|
_ = validator.eval_json(validator.stats)
|
|
145
145
|
|
|
146
|
-
args = {"model": "
|
|
146
|
+
args = {"model": "yolo26n-pose.pt", "data": "coco8-pose.yaml", "save_json": True, "imgsz": 64}
|
|
147
147
|
validator = PoseValidator(args=args)
|
|
148
148
|
validator()
|
|
149
149
|
validator.is_coco = True
|