dgenerate-ultralytics-headless 8.3.218__py3-none-any.whl → 8.3.221__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.218.dist-info → dgenerate_ultralytics_headless-8.3.221.dist-info}/METADATA +1 -1
- {dgenerate_ultralytics_headless-8.3.218.dist-info → dgenerate_ultralytics_headless-8.3.221.dist-info}/RECORD +77 -77
- tests/__init__.py +5 -7
- tests/conftest.py +3 -7
- tests/test_cli.py +9 -2
- tests/test_engine.py +1 -1
- tests/test_exports.py +37 -9
- tests/test_integrations.py +4 -4
- tests/test_python.py +37 -44
- tests/test_solutions.py +154 -145
- ultralytics/__init__.py +1 -1
- ultralytics/cfg/__init__.py +7 -5
- ultralytics/cfg/default.yaml +1 -1
- ultralytics/data/__init__.py +4 -4
- ultralytics/data/augment.py +10 -10
- ultralytics/data/base.py +1 -1
- ultralytics/data/build.py +1 -1
- ultralytics/data/converter.py +3 -3
- ultralytics/data/dataset.py +3 -3
- ultralytics/data/loaders.py +2 -2
- ultralytics/data/utils.py +2 -2
- ultralytics/engine/exporter.py +73 -20
- ultralytics/engine/model.py +1 -1
- ultralytics/engine/predictor.py +1 -0
- ultralytics/engine/trainer.py +5 -3
- ultralytics/engine/tuner.py +4 -4
- ultralytics/hub/__init__.py +9 -7
- ultralytics/hub/utils.py +2 -2
- ultralytics/models/__init__.py +1 -1
- ultralytics/models/fastsam/__init__.py +1 -1
- ultralytics/models/fastsam/predict.py +10 -16
- ultralytics/models/nas/__init__.py +1 -1
- ultralytics/models/rtdetr/__init__.py +1 -1
- ultralytics/models/sam/__init__.py +1 -1
- ultralytics/models/sam/amg.py +2 -2
- ultralytics/models/sam/modules/blocks.py +1 -1
- ultralytics/models/sam/modules/transformer.py +1 -1
- ultralytics/models/sam/predict.py +1 -1
- ultralytics/models/yolo/__init__.py +1 -1
- ultralytics/models/yolo/pose/__init__.py +1 -1
- ultralytics/models/yolo/segment/val.py +1 -1
- ultralytics/models/yolo/yoloe/__init__.py +7 -7
- ultralytics/nn/__init__.py +7 -7
- ultralytics/nn/autobackend.py +32 -5
- ultralytics/nn/modules/__init__.py +60 -60
- ultralytics/nn/modules/block.py +26 -26
- ultralytics/nn/modules/conv.py +7 -7
- ultralytics/nn/modules/head.py +1 -1
- ultralytics/nn/modules/transformer.py +7 -7
- ultralytics/nn/modules/utils.py +1 -1
- ultralytics/nn/tasks.py +3 -3
- ultralytics/solutions/__init__.py +12 -12
- ultralytics/solutions/object_counter.py +3 -6
- ultralytics/solutions/queue_management.py +1 -1
- ultralytics/solutions/similarity_search.py +3 -3
- ultralytics/trackers/__init__.py +1 -1
- ultralytics/trackers/byte_tracker.py +2 -2
- ultralytics/trackers/utils/matching.py +1 -1
- ultralytics/utils/__init__.py +2 -2
- ultralytics/utils/benchmarks.py +4 -4
- ultralytics/utils/callbacks/comet.py +2 -2
- ultralytics/utils/checks.py +2 -2
- ultralytics/utils/downloads.py +2 -2
- ultralytics/utils/export/__init__.py +1 -1
- ultralytics/utils/files.py +1 -1
- ultralytics/utils/git.py +1 -1
- ultralytics/utils/logger.py +1 -1
- ultralytics/utils/metrics.py +13 -9
- ultralytics/utils/ops.py +8 -8
- ultralytics/utils/plotting.py +2 -1
- ultralytics/utils/torch_utils.py +5 -4
- ultralytics/utils/triton.py +2 -2
- ultralytics/utils/tuner.py +4 -2
- {dgenerate_ultralytics_headless-8.3.218.dist-info → dgenerate_ultralytics_headless-8.3.221.dist-info}/WHEEL +0 -0
- {dgenerate_ultralytics_headless-8.3.218.dist-info → dgenerate_ultralytics_headless-8.3.221.dist-info}/entry_points.txt +0 -0
- {dgenerate_ultralytics_headless-8.3.218.dist-info → dgenerate_ultralytics_headless-8.3.221.dist-info}/licenses/LICENSE +0 -0
- {dgenerate_ultralytics_headless-8.3.218.dist-info → dgenerate_ultralytics_headless-8.3.221.dist-info}/top_level.txt +0 -0
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
Metadata-Version: 2.4
|
|
2
2
|
Name: dgenerate-ultralytics-headless
|
|
3
|
-
Version: 8.3.
|
|
3
|
+
Version: 8.3.221
|
|
4
4
|
Summary: Automatically built Ultralytics package with python-opencv-headless dependency instead of python-opencv
|
|
5
5
|
Author-email: Glenn Jocher <glenn.jocher@ultralytics.com>, Jing Qiu <jing.qiu@ultralytics.com>
|
|
6
6
|
Maintainer-email: Ultralytics <hello@ultralytics.com>
|
|
@@ -1,19 +1,19 @@
|
|
|
1
|
-
dgenerate_ultralytics_headless-8.3.
|
|
2
|
-
tests/__init__.py,sha256=
|
|
3
|
-
tests/conftest.py,sha256=
|
|
4
|
-
tests/test_cli.py,sha256=
|
|
1
|
+
dgenerate_ultralytics_headless-8.3.221.dist-info/licenses/LICENSE,sha256=DZak_2itbUtvHzD3E7GNUYSRK6jdOJ-GqncQ2weavLA,34523
|
|
2
|
+
tests/__init__.py,sha256=bCox_hLdGRFYGLb2kd722VdNP2zEXNYNuLLYtqZSrbw,804
|
|
3
|
+
tests/conftest.py,sha256=oaqn_-8LH7R4YQAKKOiK3iuAfjmmLQ9-pL_IPj6xq-U,2333
|
|
4
|
+
tests/test_cli.py,sha256=zygPlaksok7Nwugp3aIudDSkOlzISvmDWfKNmpY3mSA,5844
|
|
5
5
|
tests/test_cuda.py,sha256=6zUSwu3xaYiO3RRNyDkNsuyeq47b1e9f6JNhPZVeDL4,8142
|
|
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
|
|
6
|
+
tests/test_engine.py,sha256=ER2DsHM0GfUG99AH1Q-Lpm4x36qxkfOzxmH6uYM75ds,5722
|
|
7
|
+
tests/test_exports.py,sha256=OMLio2uUhyqo8D8qB5xUwmk7Po2rMeAACRc8WYoxbj4,13147
|
|
8
|
+
tests/test_integrations.py,sha256=6QgSh9n0J04RdUYz08VeVOnKmf4S5MDEQ0chzS7jo_c,6220
|
|
9
|
+
tests/test_python.py,sha256=E5Jft-fJ2S6TFEWHkNWEsZ8N3fp6ap0Dpa9zHWCSsMo,27735
|
|
10
|
+
tests/test_solutions.py,sha256=j_PZZ5tMR1Y5ararY-OTXZr1hYJ7vEVr8H3w4O1tbQs,14153
|
|
11
|
+
ultralytics/__init__.py,sha256=C9rXBpAg_ChwEIbZd1rcopKDZ6FEAK3LzWx_9Tv-quI,1302
|
|
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=jqOp7PCEN7vc-ItBiPFne3k8jagkXnL_Ep-IryuyhxA,40017
|
|
16
|
+
ultralytics/cfg/default.yaml,sha256=KKENSHolDSto1HJVGjBvTXvz9ae-XMcYRzKrjU3QfZc,8912
|
|
17
17
|
ultralytics/cfg/datasets/Argoverse.yaml,sha256=J4ItoUlE_EiYTmp1DFKYHfbqHkj8j4wUtRJQhaMIlBM,3275
|
|
18
18
|
ultralytics/cfg/datasets/DOTAv1.5.yaml,sha256=VZ_KKFX0H2YvlFVJ8JHcLWYBZ2xiQ6Z-ROSTiKWpS7c,1211
|
|
19
19
|
ultralytics/cfg/datasets/DOTAv1.yaml,sha256=JrDuYcQ0JU9lJlCA-dCkMNko_jaj6MAVGHjsfjeZ_u0,1181
|
|
@@ -106,67 +106,67 @@ ultralytics/cfg/models/v9/yolov9s.yaml,sha256=j_v3JWaPtiuM8aKJt15Z_4HPRCoHWn_G6Z
|
|
|
106
106
|
ultralytics/cfg/models/v9/yolov9t.yaml,sha256=Q8GpSXE7fumhuJiQg4a2SkuS_UmnXqp-eoZxW_C0vEo,1375
|
|
107
107
|
ultralytics/cfg/trackers/botsort.yaml,sha256=tRxC-qT4Wz0mLn5x7ZEwrqgGKrmTDVY7gMge-mhpe7U,1431
|
|
108
108
|
ultralytics/cfg/trackers/bytetrack.yaml,sha256=7LS1ObP5u7BUFcmeY6L2m3bRuPUktnpJspFKd_ElVWc,908
|
|
109
|
-
ultralytics/data/__init__.py,sha256=
|
|
109
|
+
ultralytics/data/__init__.py,sha256=ToR8zl0JhBHy42ZvV7zIwO_F3lbi5oNlGQNPK3dlddU,644
|
|
110
110
|
ultralytics/data/annotator.py,sha256=f15TCDEM8SuuzHiFB8oyhTy9vfywKmPTLSPAgsZQP9I,2990
|
|
111
|
-
ultralytics/data/augment.py,sha256=
|
|
112
|
-
ultralytics/data/base.py,sha256=
|
|
113
|
-
ultralytics/data/build.py,sha256=
|
|
114
|
-
ultralytics/data/converter.py,sha256=
|
|
115
|
-
ultralytics/data/dataset.py,sha256=
|
|
116
|
-
ultralytics/data/loaders.py,sha256=
|
|
111
|
+
ultralytics/data/augment.py,sha256=XGnatX9V8mdTCeVRvFt6I_NVng02N9-sWQgqHZd57Mk,132918
|
|
112
|
+
ultralytics/data/base.py,sha256=wf6c7Inhfz2kOTn-y5QvS8iPulnPSRMws06kE2Z4ipo,19662
|
|
113
|
+
ultralytics/data/build.py,sha256=a-Gs2JbZS9k59CpJaax7kby-PuZOeK1NYV4mq4KRFA0,16688
|
|
114
|
+
ultralytics/data/converter.py,sha256=EPVgUY8que3TaQo6Sxx-OOUIZulVNBaw1MIdGBUZECs,31963
|
|
115
|
+
ultralytics/data/dataset.py,sha256=HQ3r6Slu4LHSNXnRCiOgbs2zC2E2uAQkYnHZhV3JGdE,36772
|
|
116
|
+
ultralytics/data/loaders.py,sha256=lTnKDRtwZ9IRpZbH2BO8WVHlDuKeTp5ExVBvbUDbdMA,31732
|
|
117
117
|
ultralytics/data/split.py,sha256=5ubnL_wsEutFQOj4I4K01L9UpZrrO_vO3HrydSLJyIY,5107
|
|
118
118
|
ultralytics/data/split_dota.py,sha256=Lz04qVufTvHn4cTyo3VkqoIM93rb-Ymr8uOIXeSsaJI,12910
|
|
119
|
-
ultralytics/data/utils.py,sha256=
|
|
119
|
+
ultralytics/data/utils.py,sha256=7_HpSo4tF9ToRzG9EQXhuw4GCusGPOXBmWGw3h6Gp_A,36931
|
|
120
120
|
ultralytics/data/scripts/download_weights.sh,sha256=0y8XtZxOru7dVThXDFUXLHBuICgOIqZNUwpyL4Rh6lg,595
|
|
121
121
|
ultralytics/data/scripts/get_coco.sh,sha256=UuJpJeo3qQpTHVINeOpmP0NYmg8PhEFE3A8J3jKrnPw,1768
|
|
122
122
|
ultralytics/data/scripts/get_coco128.sh,sha256=qmRQl_hOKrsdHrTrnyQuFIH01oDz3lfaz138OgGfLt8,650
|
|
123
123
|
ultralytics/data/scripts/get_imagenet.sh,sha256=hr42H16bM47iT27rgS7MpEo-GeOZAYUQXgr0B2cwn48,1705
|
|
124
124
|
ultralytics/engine/__init__.py,sha256=lm6MckFYCPTbqIoX7w0s_daxdjNeBeKW6DXppv1-QUM,70
|
|
125
|
-
ultralytics/engine/exporter.py,sha256=
|
|
126
|
-
ultralytics/engine/model.py,sha256=
|
|
127
|
-
ultralytics/engine/predictor.py,sha256=
|
|
125
|
+
ultralytics/engine/exporter.py,sha256=dFbnKY_6Ym0aY6gk1GYspgGTHI6UA6LSLUJaxAAwPDU,73543
|
|
126
|
+
ultralytics/engine/model.py,sha256=sCa5W6Qit5NloEVuuvNelH-geJQw9IGjcOc9_PXNDnI,53456
|
|
127
|
+
ultralytics/engine/predictor.py,sha256=ZQrx1Bz4X8aTgGjrOSdRSP7SCtQ05uqz6IitEan_Gyk,22813
|
|
128
128
|
ultralytics/engine/results.py,sha256=oHQdV_eIMvAU2qLCV7wG7iLifdfaLEgP80lDXB5ghkg,71490
|
|
129
|
-
ultralytics/engine/trainer.py,sha256=
|
|
130
|
-
ultralytics/engine/tuner.py,sha256=
|
|
129
|
+
ultralytics/engine/trainer.py,sha256=crS7HL4sLOeXn_51_2mKBfeC5jfR6GeT-1qNgblD7dQ,44140
|
|
130
|
+
ultralytics/engine/tuner.py,sha256=vc_Y26g5vzuIeSQBUPOwyVWNT6DumpEDGLvYygdq_kY,21659
|
|
131
131
|
ultralytics/engine/validator.py,sha256=HoBE5Lc7Ro7IQ5TlfDC94L5BmdOb8J8yfGlELozZ6EM,17560
|
|
132
|
-
ultralytics/hub/__init__.py,sha256=
|
|
132
|
+
ultralytics/hub/__init__.py,sha256=2BzU31EQXxnRL9r3BgUFsskf_o_FgTBTi4az58frNNM,6741
|
|
133
133
|
ultralytics/hub/auth.py,sha256=RIwZDWfW6vS2yGpZKR0xVl0-38itJYEFtmqY_M70bl8,6304
|
|
134
134
|
ultralytics/hub/session.py,sha256=1o9vdd_fvPUHQ5oZgljtPePuPMUalIoXqOvE7Sdmd2o,18450
|
|
135
|
-
ultralytics/hub/utils.py,sha256=
|
|
135
|
+
ultralytics/hub/utils.py,sha256=bTuYT1TolGKUYUlz5nxfVFEjr2T6PKV3OLY6fkM65mE,6368
|
|
136
136
|
ultralytics/hub/google/__init__.py,sha256=8o3RorFafO_DzlzImXnzNQXtyPM1k-CQ8tsWSfnMMac,8468
|
|
137
|
-
ultralytics/models/__init__.py,sha256=
|
|
138
|
-
ultralytics/models/fastsam/__init__.py,sha256=
|
|
137
|
+
ultralytics/models/__init__.py,sha256=ljus_u1CIuP99k9fu6sCtzIeFZ-TCE28NZ8kefZHFNY,309
|
|
138
|
+
ultralytics/models/fastsam/__init__.py,sha256=Ku89Fy_X8ok3YPEUajjUZ5i4O08jdJMjJHt-3Z99Frk,231
|
|
139
139
|
ultralytics/models/fastsam/model.py,sha256=vIdl536LUrefjqMFEJ-9UyK4Ta6p2ki2G_gn2DZ9X_Y,3438
|
|
140
|
-
ultralytics/models/fastsam/predict.py,sha256=
|
|
140
|
+
ultralytics/models/fastsam/predict.py,sha256=QoG9GnUKVNIDgxeVMIFSP3YbOOFcqVvaQiGaUiAoFzc,8664
|
|
141
141
|
ultralytics/models/fastsam/utils.py,sha256=yuCXB4CVjRx8lDf61DP8B6qMx7TVf7AynQvdWREeFco,884
|
|
142
142
|
ultralytics/models/fastsam/val.py,sha256=oLxB8vBKTfiT7eBbTzvpqq_xNSvDOjGdP1J7egHGsCA,2041
|
|
143
|
-
ultralytics/models/nas/__init__.py,sha256=
|
|
143
|
+
ultralytics/models/nas/__init__.py,sha256=Q4ZQak8xNWtV5YSw_pFu0anbCyDxxEAuMMDfMzu6-0s,207
|
|
144
144
|
ultralytics/models/nas/model.py,sha256=Z2Mq4uiI9Mk2qYLFha5j3efpHVuJ5ySfpdAu9kFGPDc,3903
|
|
145
145
|
ultralytics/models/nas/predict.py,sha256=J4UT7nwi_h63lJ3a_gYac-Ws8wFYingZINxMqSoaX5E,2706
|
|
146
146
|
ultralytics/models/nas/val.py,sha256=QUTE3zuhJLVqmDGd2n7iSSk7X6jKZCRxufFkBbyxYYo,1548
|
|
147
|
-
ultralytics/models/rtdetr/__init__.py,sha256=
|
|
147
|
+
ultralytics/models/rtdetr/__init__.py,sha256=F4NEQqtcVKFxj97Dh7rkn2Vu3JG4Ea_nxqrBB-9P1vc,225
|
|
148
148
|
ultralytics/models/rtdetr/model.py,sha256=Pq9QDgaZetDnjxdYSoomj2s6vOGSdpsqVfyN5j0GUmc,2292
|
|
149
149
|
ultralytics/models/rtdetr/predict.py,sha256=43-gGCHEH7UQQ6H1oXdlDlrM39esnp-YEhqCvZOwtOM,4279
|
|
150
150
|
ultralytics/models/rtdetr/train.py,sha256=SNntxGHXatbNqn1yna5_dDQiR_ciDK6o_4S7JIHU7EY,3765
|
|
151
151
|
ultralytics/models/rtdetr/val.py,sha256=l26CzpcYHYC0sQ--rKUFBCYl73nsgAGOj1U3xScNzFs,8918
|
|
152
|
-
ultralytics/models/sam/__init__.py,sha256=
|
|
153
|
-
ultralytics/models/sam/amg.py,sha256=
|
|
152
|
+
ultralytics/models/sam/__init__.py,sha256=p1BKLawQFvVxmdk7LomFVWX-67Kc-AP4PJBNPfU_Nuc,359
|
|
153
|
+
ultralytics/models/sam/amg.py,sha256=nFq4EwHf65W2N5Ipo4W69nGRhCbJEh_boYQ8SIPWBZ0,11816
|
|
154
154
|
ultralytics/models/sam/build.py,sha256=uKCgHpcYgV26OFuMq5RaGR8aXYoEtNoituT06bmnW44,12790
|
|
155
155
|
ultralytics/models/sam/model.py,sha256=qV8tlHQA1AHUqGkWbwtI7cLw0Rgy3a4X9S2c_wu5fh4,7237
|
|
156
|
-
ultralytics/models/sam/predict.py,sha256=
|
|
156
|
+
ultralytics/models/sam/predict.py,sha256=0jE1YOKxfmdlPXk-ZTcEJgRnXHrJqMCN90hgLawJsbc,105178
|
|
157
157
|
ultralytics/models/sam/modules/__init__.py,sha256=lm6MckFYCPTbqIoX7w0s_daxdjNeBeKW6DXppv1-QUM,70
|
|
158
|
-
ultralytics/models/sam/modules/blocks.py,sha256=
|
|
158
|
+
ultralytics/models/sam/modules/blocks.py,sha256=IfkQMyrUXzeInkynRdVtpCW70l_wvDNIR7SC9nV2YyQ,46062
|
|
159
159
|
ultralytics/models/sam/modules/decoders.py,sha256=PGNNpy1ttAy6xV_ERW1Ld3Kf9LGDG3mibOss0SeHAis,25623
|
|
160
160
|
ultralytics/models/sam/modules/encoders.py,sha256=VOgwSDFep_zqssESz8mNDPDdJfQmP97kHVN-MrExGnk,37326
|
|
161
161
|
ultralytics/models/sam/modules/memory_attention.py,sha256=BOkV6ULHc0Iiw_tHcNYosYrZ1tAXyC0DG46ktQzR91E,13638
|
|
162
162
|
ultralytics/models/sam/modules/sam.py,sha256=6GuhW7nGyNfyD1p6DT804gy8mFGIrzpsV-4SrqJXQnw,55641
|
|
163
163
|
ultralytics/models/sam/modules/tiny_encoder.py,sha256=fSxTByC7OSmHYg93KylsFayh6nPdlidRk1BORh6X-p0,42199
|
|
164
|
-
ultralytics/models/sam/modules/transformer.py,sha256=
|
|
164
|
+
ultralytics/models/sam/modules/transformer.py,sha256=SN4p7TqwfmCYjEpnrV3xZ2ObG_qigkW5DlvxjTtRpTM,14965
|
|
165
165
|
ultralytics/models/sam/modules/utils.py,sha256=XReheR5K0jbTKYy5k_iSC1vocUndi8aBkesz-n6Pl9g,16045
|
|
166
166
|
ultralytics/models/utils/__init__.py,sha256=lm6MckFYCPTbqIoX7w0s_daxdjNeBeKW6DXppv1-QUM,70
|
|
167
167
|
ultralytics/models/utils/loss.py,sha256=NABWnevvc9eMYKqo1m2f-lLICFavZQITyNlPbcX1xi4,21231
|
|
168
168
|
ultralytics/models/utils/ops.py,sha256=HkIrCE0wTiXPmHCDM8IMAy0inOy7U6ZABWqu5_KY0qo,15239
|
|
169
|
-
ultralytics/models/yolo/__init__.py,sha256=
|
|
169
|
+
ultralytics/models/yolo/__init__.py,sha256=YD407NDDiyjo0x_MR6usJaTpePKPgsfBUYehlCw7lRs,307
|
|
170
170
|
ultralytics/models/yolo/model.py,sha256=PH8nXl0ZulgjWMr9M-XAK2TcdaBNXX5AzofIhcKbTQ0,18840
|
|
171
171
|
ultralytics/models/yolo/classify/__init__.py,sha256=9--HVaNOfI1K7rn_rRqclL8FUAnpfeBrRqEQIaQw2xM,383
|
|
172
172
|
ultralytics/models/yolo/classify/predict.py,sha256=o7pDE8xwjkHUUIIOph7ZVQZyGZyob24dYDQ460v_7R0,4149
|
|
@@ -180,34 +180,34 @@ ultralytics/models/yolo/obb/__init__.py,sha256=tQmpG8wVHsajWkZdmD6cjGohJ4ki64iSX
|
|
|
180
180
|
ultralytics/models/yolo/obb/predict.py,sha256=4r1eSld6TNJlk9JG56e-DX6oPL8uBBqiuztyBpxWlHE,2888
|
|
181
181
|
ultralytics/models/yolo/obb/train.py,sha256=BbehrsKP0lHRV3v7rrw8wAeiDdc-szbhHAmDy0OdhoM,3461
|
|
182
182
|
ultralytics/models/yolo/obb/val.py,sha256=9jMnBRIqPkCzY21CSiuP3LL4qpBEY-pnEgKQSi4bEJ0,14187
|
|
183
|
-
ultralytics/models/yolo/pose/__init__.py,sha256=
|
|
183
|
+
ultralytics/models/yolo/pose/__init__.py,sha256=_9OFLj19XwvJHBRxQtVW5CV7rvJ_3hDPE97miit0sPc,227
|
|
184
184
|
ultralytics/models/yolo/pose/predict.py,sha256=3fgu4EKcVRKlP7fySDVsngl4ufk2f71P8SLbfRU2KgE,3747
|
|
185
185
|
ultralytics/models/yolo/pose/train.py,sha256=bR-TfahC0vc9AM_bOg5HhClgaNECzIWPFtu8GNjg180,4958
|
|
186
186
|
ultralytics/models/yolo/pose/val.py,sha256=MK-GueXmXrl7eZ5WHYjJMghE4AYJTEut7AuS-G5D1gw,12650
|
|
187
187
|
ultralytics/models/yolo/segment/__init__.py,sha256=3IThhZ1wlkY9FvmWm9cE-5-ZyE6F1FgzAtQ6jOOFzzw,275
|
|
188
188
|
ultralytics/models/yolo/segment/predict.py,sha256=Qf6B4v2O8usK5wHfbre4gkJjEWKidxZRhetWv4nyr6M,5470
|
|
189
189
|
ultralytics/models/yolo/segment/train.py,sha256=5aPK5FDHLzbXb3R5TCpsAr1O6-8rtupOIoDokY8bSDs,3032
|
|
190
|
-
ultralytics/models/yolo/segment/val.py,sha256=
|
|
190
|
+
ultralytics/models/yolo/segment/val.py,sha256=4NO0k32Un-gJCE153PT1w-E82X14cfgSgTIiT_YqpRo,11305
|
|
191
191
|
ultralytics/models/yolo/world/__init__.py,sha256=nlh8I6t8hMGz_vZg8QSlsUW1R-2eKvn9CGUoPPQEGhA,131
|
|
192
192
|
ultralytics/models/yolo/world/train.py,sha256=IBuzLgsNJEFuMaWgrhE3sqIl0vltdzxlPj9Wm0S2diI,7956
|
|
193
193
|
ultralytics/models/yolo/world/train_world.py,sha256=9p9YIckrATaJjGOrpmuC8MbZX9qdoCPCEV9EGZ0sExg,9553
|
|
194
|
-
ultralytics/models/yolo/yoloe/__init__.py,sha256=
|
|
194
|
+
ultralytics/models/yolo/yoloe/__init__.py,sha256=zaZo1_ommaxNv7mD7xpdSomNF4s8mpOcCVTXspg0ncY,760
|
|
195
195
|
ultralytics/models/yolo/yoloe/predict.py,sha256=pcbAUbosr1Xc436MfQi6ah3MQ6kkPzjOcltmdA3VMDE,7124
|
|
196
196
|
ultralytics/models/yolo/yoloe/train.py,sha256=qefvNNXDTOK1tO3va0kNHr8lE5QJkOlV8GdZdRx3Mis,13034
|
|
197
197
|
ultralytics/models/yolo/yoloe/train_seg.py,sha256=aCV7M8oQOvODFnU4piZdJh3tIrBJYAzZfRVRx1vRgxo,4956
|
|
198
198
|
ultralytics/models/yolo/yoloe/val.py,sha256=5Gd9EoFH0FmKKvWXBl4J7gBe9DVxIczN-s3ceHwdUDo,9458
|
|
199
|
-
ultralytics/nn/__init__.py,sha256=
|
|
200
|
-
ultralytics/nn/autobackend.py,sha256=
|
|
201
|
-
ultralytics/nn/tasks.py,sha256=
|
|
199
|
+
ultralytics/nn/__init__.py,sha256=538LZPUKKvc3JCMgiQ4VLGqRN2ZAaVLFcQbeNNHFkEA,545
|
|
200
|
+
ultralytics/nn/autobackend.py,sha256=918iNweM3fTuRIbxHbXC1wspOj9rlGkuwalQ61uYLbk,42694
|
|
201
|
+
ultralytics/nn/tasks.py,sha256=vRr6HTucM7Eg3kxzhYtyjgEAdacZ7gIDU3yPbMnyYMM,70834
|
|
202
202
|
ultralytics/nn/text_model.py,sha256=pHqnKe8UueR1MuwJcIE_IvrnYIlt68QL796xjcRJs2A,15275
|
|
203
|
-
ultralytics/nn/modules/__init__.py,sha256=
|
|
203
|
+
ultralytics/nn/modules/__init__.py,sha256=5Sg_28MDfKwdu14Ty_WCaiIXZyjBSQ-xCNCwnoz_w-w,3198
|
|
204
204
|
ultralytics/nn/modules/activation.py,sha256=75JcIMH2Cu9GTC2Uf55r_5YLpxcrXQDaVoeGQ0hlUAU,2233
|
|
205
|
-
ultralytics/nn/modules/block.py,sha256
|
|
206
|
-
ultralytics/nn/modules/conv.py,sha256=
|
|
207
|
-
ultralytics/nn/modules/head.py,sha256=
|
|
208
|
-
ultralytics/nn/modules/transformer.py,sha256=
|
|
209
|
-
ultralytics/nn/modules/utils.py,sha256=
|
|
210
|
-
ultralytics/solutions/__init__.py,sha256=
|
|
205
|
+
ultralytics/nn/modules/block.py,sha256=eQ8DegyvBG9k-O_QgSZe5XGmpravqwlnSCCBW6bHRXo,70622
|
|
206
|
+
ultralytics/nn/modules/conv.py,sha256=MISNAK8NzAZhNUusVKWvTHQ8IsofwM-5X0gChCagsaY,21457
|
|
207
|
+
ultralytics/nn/modules/head.py,sha256=HBSoHOcd9hikpBLF9BdgIItYJDNWIO8NiNYMcMw6ThM,53512
|
|
208
|
+
ultralytics/nn/modules/transformer.py,sha256=9aq0Yo9V3C4y_McSje4qE1d_PTWDctTsrb98MyXxigc,31470
|
|
209
|
+
ultralytics/nn/modules/utils.py,sha256=9kLeEtvEBFLugz53TkdI4mifD-39a-upjPD-wrE8opU,6092
|
|
210
|
+
ultralytics/solutions/__init__.py,sha256=Jj7OcRiYjHH-e104H4xTgjjR5W6aPB4mBRndbaSPmgU,1209
|
|
211
211
|
ultralytics/solutions/ai_gym.py,sha256=VHUYkq2AT5Zaee-Px9abvN97thhomz7VDqg0HNZLKLI,5217
|
|
212
212
|
ultralytics/solutions/analytics.py,sha256=8pGHU6qsTUNuSF0c8rZbANsdYoVDCv6-Vyevb9zdE-s,12823
|
|
213
213
|
ultralytics/solutions/config.py,sha256=-VovlZFb-2VXAhoyEgK-A1FHECWEk4PlQrqeQArC4UY,5396
|
|
@@ -215,57 +215,57 @@ ultralytics/solutions/distance_calculation.py,sha256=_TExUABsEgS-c_WbUP0dK5JxIcA
|
|
|
215
215
|
ultralytics/solutions/heatmap.py,sha256=5rS7FjlNQlokOpkdpJFZfXFzMiIV-h2mr4Jp0wNogo4,5545
|
|
216
216
|
ultralytics/solutions/instance_segmentation.py,sha256=GWJQmnHkSO_S5dvKMI2NLMejk9h9y0G6DEiXTGUUPm4,3789
|
|
217
217
|
ultralytics/solutions/object_blurrer.py,sha256=96KOAEagk4UoErlUMiIDK6j1CWs2nN1dcJ5V6pl9L-8,3992
|
|
218
|
-
ultralytics/solutions/object_counter.py,sha256=
|
|
218
|
+
ultralytics/solutions/object_counter.py,sha256=B031BQ1QZCB28exffdBE2DLxq3BUNMoQVTGm3zGpcUg,9387
|
|
219
219
|
ultralytics/solutions/object_cropper.py,sha256=lRKtWINAe9GDxau1Xejbjydsqg2hrpGZXPtZwTgvyKQ,3603
|
|
220
220
|
ultralytics/solutions/parking_management.py,sha256=DMPl1rd0TegTrUvrCM44_y-HZTx3DVuEG6W9BC7lR0E,13636
|
|
221
|
-
ultralytics/solutions/queue_management.py,sha256=
|
|
221
|
+
ultralytics/solutions/queue_management.py,sha256=pSYyJP45vDcaKRHTAFgXFbhfBaiJdtn-24b6jKV3wtw,4385
|
|
222
222
|
ultralytics/solutions/region_counter.py,sha256=KjU5nErQ_maNzchtS3Cu54PcGTf_yxaR8iBZwFRSPNI,6048
|
|
223
223
|
ultralytics/solutions/security_alarm.py,sha256=czEaMcy04q-iBkKqT_14d8H20CFB6zcKH_31nBGQnyw,6345
|
|
224
|
-
ultralytics/solutions/similarity_search.py,sha256=
|
|
224
|
+
ultralytics/solutions/similarity_search.py,sha256=UbC8jiR6OaAd9senm1DmhorZjDaHl_ngMYrXbLFHJLI,9681
|
|
225
225
|
ultralytics/solutions/solutions.py,sha256=syChH-uYq6YGspXflKJF96gNVnkxOLobkLM_ceMZI6Q,36042
|
|
226
226
|
ultralytics/solutions/speed_estimation.py,sha256=chg_tBuKFw3EnFiv_obNDaUXLAo-FypxC7gsDeB_VUI,5878
|
|
227
227
|
ultralytics/solutions/streamlit_inference.py,sha256=28p2fBTsCLtN7jOv4Q2Ppw3BHwhowsmWV5y82iXC8WU,13074
|
|
228
228
|
ultralytics/solutions/trackzone.py,sha256=6W_55Iio884FCj12r5zItAkedStAnTfz3ZNEYxQ7ozg,3941
|
|
229
229
|
ultralytics/solutions/vision_eye.py,sha256=GiooS_ajmhafjqlAGENEDsGPKsqmThq9mHrzuHHeghg,3005
|
|
230
230
|
ultralytics/solutions/templates/similarity-search.html,sha256=nyyurpWlkvYlDeNh-74TlV4ctCpTksvkVy2Yc4ImQ1U,4261
|
|
231
|
-
ultralytics/trackers/__init__.py,sha256=
|
|
231
|
+
ultralytics/trackers/__init__.py,sha256=n3BOO0TR-Sz5ANDYOkKDipM9nSHOePMEwqafbk-YEPs,255
|
|
232
232
|
ultralytics/trackers/basetrack.py,sha256=-skBFFatzgJFAPN9Frm1u1h_RDUg3WOlxG6eHQxp2Gw,4384
|
|
233
233
|
ultralytics/trackers/bot_sort.py,sha256=IiC1MSP2He2FTSl6u5BBImVjLwAtq8g7hkeOroTEnCk,12255
|
|
234
|
-
ultralytics/trackers/byte_tracker.py,sha256=
|
|
234
|
+
ultralytics/trackers/byte_tracker.py,sha256=2GQ3ftXgaxZijYfDdp-LLnyNZ6DDQspBBAJdN8yY61Y,21494
|
|
235
235
|
ultralytics/trackers/track.py,sha256=MHMydDt_MfXdj6naO2lLuEPF46pZUbDmz5Sqtr18-J4,4757
|
|
236
236
|
ultralytics/trackers/utils/__init__.py,sha256=lm6MckFYCPTbqIoX7w0s_daxdjNeBeKW6DXppv1-QUM,70
|
|
237
237
|
ultralytics/trackers/utils/gmc.py,sha256=1cCmlbk5Z6Pd-rFCaiJML7o_cUm_IktMuCocTDOMGFQ,14028
|
|
238
238
|
ultralytics/trackers/utils/kalman_filter.py,sha256=PPmM0lwBMdT_hGojvfLoUsBUFMBBMNRAxKbMcQa3wJ0,21619
|
|
239
|
-
ultralytics/trackers/utils/matching.py,sha256=
|
|
240
|
-
ultralytics/utils/__init__.py,sha256=
|
|
239
|
+
ultralytics/trackers/utils/matching.py,sha256=eIcqzN6WgvE8A3V83DlyklcWLN386StVieqB-j35TpM,7164
|
|
240
|
+
ultralytics/utils/__init__.py,sha256=3NyzCFKTea_9pt7eKRNCp6h0SyFJcEYp3s4bdVJqnwc,53491
|
|
241
241
|
ultralytics/utils/autobatch.py,sha256=i6KYLLSItKP1Q2IUlTPHrZhjcxl7UOjs0Seb8bF8pvM,5124
|
|
242
242
|
ultralytics/utils/autodevice.py,sha256=d9yq6eEn05fdfzfpxeSECd0YEO61er5f7T-0kjLdofg,8843
|
|
243
|
-
ultralytics/utils/benchmarks.py,sha256
|
|
244
|
-
ultralytics/utils/checks.py,sha256=
|
|
243
|
+
ultralytics/utils/benchmarks.py,sha256=-JP214Raik0L9pko-D_PLJwU_PZsNNuEuYLsZ3LJNTQ,32124
|
|
244
|
+
ultralytics/utils/checks.py,sha256=QyQglwJBi-SYA0lomfq6FFvkvOuyd9Vz_qmz0SMdHII,36272
|
|
245
245
|
ultralytics/utils/cpu.py,sha256=OPlVxROWhQp-kEa9EkeNRKRQ-jz0KwySu5a-h91JZjk,3634
|
|
246
246
|
ultralytics/utils/dist.py,sha256=5xQhWK0OLORvseAL08UmG1LYdkiDVLquxmaGSnqiSqo,4151
|
|
247
|
-
ultralytics/utils/downloads.py,sha256=
|
|
247
|
+
ultralytics/utils/downloads.py,sha256=BAAjICG0ib4cifoTRS0D59uRXOBhXwY6fT-eUdEZgFU,22933
|
|
248
248
|
ultralytics/utils/errors.py,sha256=XT9Ru7ivoBgofK6PlnyigGoa7Fmf5nEhyHtnD-8TRXI,1584
|
|
249
249
|
ultralytics/utils/events.py,sha256=v2RmLlx78_K6xQfOAuUTJMOexAgNdiuiOvvnsH65oDA,4679
|
|
250
|
-
ultralytics/utils/files.py,sha256=
|
|
251
|
-
ultralytics/utils/git.py,sha256=
|
|
250
|
+
ultralytics/utils/files.py,sha256=DnaeaGM073D6lEc9G5fo4p1YVJeT45-ZF1XvC-EsaWQ,8205
|
|
251
|
+
ultralytics/utils/git.py,sha256=yYJH7vdWHBag1boyqYJEbwtDPdkUK4jz4cU62UY48gU,5512
|
|
252
252
|
ultralytics/utils/instance.py,sha256=_b_jMTECWJGzncCiTg7FtTDSSeXGnbiAhaJhIsqbn9k,19043
|
|
253
|
-
ultralytics/utils/logger.py,sha256=
|
|
253
|
+
ultralytics/utils/logger.py,sha256=hK1APBBHmlLAm0zbAFY7gf7Iaejy0PdwLWnnpboboGg,15129
|
|
254
254
|
ultralytics/utils/loss.py,sha256=wJ0F2DpRTI9-e9adxIm2io0zcXRa0RTWFTOc7WmS1-A,39827
|
|
255
|
-
ultralytics/utils/metrics.py,sha256=
|
|
255
|
+
ultralytics/utils/metrics.py,sha256=8D1BTd0KL-2uymHDdmxIbuni1PqDNBfZYCZTA2P3vH0,68882
|
|
256
256
|
ultralytics/utils/nms.py,sha256=AVOmPuUTEJqmq2J6rvjq-nHNxYIyabgzHdc41siyA0w,14161
|
|
257
|
-
ultralytics/utils/ops.py,sha256=
|
|
257
|
+
ultralytics/utils/ops.py,sha256=yb0jlahjxqUT_xb3y9wz0kXn0rx2AryUgWdtLat3yWY,27010
|
|
258
258
|
ultralytics/utils/patches.py,sha256=0-2G4jXCIPnMonlft-cPcjfFcOXQS6ODwUDNUwanfg4,6541
|
|
259
|
-
ultralytics/utils/plotting.py,sha256=
|
|
259
|
+
ultralytics/utils/plotting.py,sha256=kzYd5nyUklXdNiUFoHoKCRMuhxqPk1-2-hhO_twUjDw,48369
|
|
260
260
|
ultralytics/utils/tal.py,sha256=7KQYNyetfx18CNc_bvNG7BDb44CIU3DEu4qziVVvNAE,20869
|
|
261
|
-
ultralytics/utils/torch_utils.py,sha256=
|
|
261
|
+
ultralytics/utils/torch_utils.py,sha256=9o0vvqoRQ-F68yDP8xBHx8Re055u4-mNKwFSB1NN-f0,40377
|
|
262
262
|
ultralytics/utils/tqdm.py,sha256=ny5RIg2OTkWQ7gdaXfYaoIgR0Xn2_hNGB6tUpO2Unns,16137
|
|
263
|
-
ultralytics/utils/triton.py,sha256=
|
|
264
|
-
ultralytics/utils/tuner.py,sha256=
|
|
263
|
+
ultralytics/utils/triton.py,sha256=KKVCvvJiiK8XEJQYgaar5Eh0XNidqSOXMI_ng2G5MTo,5406
|
|
264
|
+
ultralytics/utils/tuner.py,sha256=MyPiqdFQVV927u9iEjKXqxWZDeJgB3rvya0HwGcE2CQ,6866
|
|
265
265
|
ultralytics/utils/callbacks/__init__.py,sha256=hzL63Rce6VkZhP4Lcim9LKjadixaQG86nKqPhk7IkS0,242
|
|
266
266
|
ultralytics/utils/callbacks/base.py,sha256=dGir0vkJY4jjprW63e23Qy4kHUT5dOINPii6HnwJuPg,6893
|
|
267
267
|
ultralytics/utils/callbacks/clearml.py,sha256=SHRtmup9BfvkZ56--BvXEoud-EMdJHY88yt973acksE,6067
|
|
268
|
-
ultralytics/utils/callbacks/comet.py,sha256
|
|
268
|
+
ultralytics/utils/callbacks/comet.py,sha256=k7AT5HBOVC9iKZSOnAL8GCiD5ANX9DO7XOcHcQDSp28,25385
|
|
269
269
|
ultralytics/utils/callbacks/dvc.py,sha256=NV0DXMQ1B5Sk5fmh60QFUGkifrAz-vwit5qhdfsyqXc,7511
|
|
270
270
|
ultralytics/utils/callbacks/hub.py,sha256=fVLqqr3ZM6hoYFlVMEeejfq1MWDrkWCskPFOG3HGILQ,4159
|
|
271
271
|
ultralytics/utils/callbacks/mlflow.py,sha256=6K8I5zij1yq3TUW9c5BBQNqdzz3IXugQjwKoBOvV6ag,5344
|
|
@@ -274,10 +274,10 @@ ultralytics/utils/callbacks/platform.py,sha256=a7T_8htoBB0uX1WIc392UJnhDjxkRyQMv
|
|
|
274
274
|
ultralytics/utils/callbacks/raytune.py,sha256=S6Bq16oQDQ8BQgnZzA0zJHGN_BBr8iAM_WtGoLiEcwg,1283
|
|
275
275
|
ultralytics/utils/callbacks/tensorboard.py,sha256=_4nfGK1dDLn6ijpvphBDhc-AS8qhS3jjY2CAWB7SNF0,5283
|
|
276
276
|
ultralytics/utils/callbacks/wb.py,sha256=ngQO8EJ1kxJDF1YajScVtzBbm26jGuejA0uWeOyvf5A,7685
|
|
277
|
-
ultralytics/utils/export/__init__.py,sha256=
|
|
277
|
+
ultralytics/utils/export/__init__.py,sha256=uyRhb-0Z5FVf7vSz2Yba1m7g5m2U_ftAv4ThlmMsqZ8,10015
|
|
278
278
|
ultralytics/utils/export/imx.py,sha256=Jl5nuNxqaP_bY5yrV2NypmoJSrexHE71TxR72SDdjcg,11394
|
|
279
|
-
dgenerate_ultralytics_headless-8.3.
|
|
280
|
-
dgenerate_ultralytics_headless-8.3.
|
|
281
|
-
dgenerate_ultralytics_headless-8.3.
|
|
282
|
-
dgenerate_ultralytics_headless-8.3.
|
|
283
|
-
dgenerate_ultralytics_headless-8.3.
|
|
279
|
+
dgenerate_ultralytics_headless-8.3.221.dist-info/METADATA,sha256=h8s7jJqH_s5Gat6HfV2r_S8KTrCIrHRpk0AH20FUXu4,38763
|
|
280
|
+
dgenerate_ultralytics_headless-8.3.221.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
|
|
281
|
+
dgenerate_ultralytics_headless-8.3.221.dist-info/entry_points.txt,sha256=YM_wiKyTe9yRrsEfqvYolNO5ngwfoL4-NwgKzc8_7sI,93
|
|
282
|
+
dgenerate_ultralytics_headless-8.3.221.dist-info/top_level.txt,sha256=XP49TwiMw4QGsvTLSYiJhz1xF_k7ev5mQ8jJXaXi45Q,12
|
|
283
|
+
dgenerate_ultralytics_headless-8.3.221.dist-info/RECORD,,
|
tests/__init__.py
CHANGED
|
@@ -1,25 +1,23 @@
|
|
|
1
1
|
# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
|
|
2
2
|
|
|
3
3
|
from ultralytics.cfg import TASK2DATA, TASK2MODEL, TASKS
|
|
4
|
-
from ultralytics.utils import ASSETS,
|
|
4
|
+
from ultralytics.utils import ASSETS, WEIGHTS_DIR, checks
|
|
5
5
|
|
|
6
6
|
# Constants used in tests
|
|
7
7
|
MODEL = WEIGHTS_DIR / "path with spaces" / "yolo11n.pt" # test spaces in path
|
|
8
8
|
CFG = "yolo11n.yaml"
|
|
9
9
|
SOURCE = ASSETS / "bus.jpg"
|
|
10
10
|
SOURCES_LIST = [ASSETS / "bus.jpg", ASSETS, ASSETS / "*", ASSETS / "**/*.jpg"]
|
|
11
|
-
TMP = (ROOT / "../tests/tmp").resolve() # temp directory for test files
|
|
12
11
|
CUDA_IS_AVAILABLE = checks.cuda_is_available()
|
|
13
12
|
CUDA_DEVICE_COUNT = checks.cuda_device_count()
|
|
14
13
|
TASK_MODEL_DATA = [(task, WEIGHTS_DIR / TASK2MODEL[task], TASK2DATA[task]) for task in TASKS]
|
|
15
|
-
MODELS = frozenset(list(TASK2MODEL.values())
|
|
14
|
+
MODELS = frozenset([*list(TASK2MODEL.values()), "yolo11n-grayscale.pt"])
|
|
16
15
|
|
|
17
16
|
__all__ = (
|
|
18
|
-
"MODEL",
|
|
19
17
|
"CFG",
|
|
18
|
+
"CUDA_DEVICE_COUNT",
|
|
19
|
+
"CUDA_IS_AVAILABLE",
|
|
20
|
+
"MODEL",
|
|
20
21
|
"SOURCE",
|
|
21
22
|
"SOURCES_LIST",
|
|
22
|
-
"TMP",
|
|
23
|
-
"CUDA_IS_AVAILABLE",
|
|
24
|
-
"CUDA_DEVICE_COUNT",
|
|
25
23
|
)
|
tests/conftest.py
CHANGED
|
@@ -3,8 +3,6 @@
|
|
|
3
3
|
import shutil
|
|
4
4
|
from pathlib import Path
|
|
5
5
|
|
|
6
|
-
from tests import TMP
|
|
7
|
-
|
|
8
6
|
|
|
9
7
|
def pytest_addoption(parser):
|
|
10
8
|
"""Add custom command-line options to pytest."""
|
|
@@ -29,7 +27,7 @@ def pytest_sessionstart(session):
|
|
|
29
27
|
Initialize session configurations for pytest.
|
|
30
28
|
|
|
31
29
|
This function is automatically called by pytest after the 'Session' object has been created but before performing
|
|
32
|
-
test collection. It sets the initial seeds
|
|
30
|
+
test collection. It sets the initial seeds for the test session.
|
|
33
31
|
|
|
34
32
|
Args:
|
|
35
33
|
session: The pytest session object.
|
|
@@ -37,8 +35,6 @@ def pytest_sessionstart(session):
|
|
|
37
35
|
from ultralytics.utils.torch_utils import init_seeds
|
|
38
36
|
|
|
39
37
|
init_seeds()
|
|
40
|
-
shutil.rmtree(TMP, ignore_errors=True) # Delete any existing tests/tmp directory
|
|
41
|
-
TMP.mkdir(parents=True, exist_ok=True) # Create a new empty directory
|
|
42
38
|
|
|
43
39
|
|
|
44
40
|
def pytest_terminal_summary(terminalreporter, exitstatus, config):
|
|
@@ -57,10 +53,10 @@ def pytest_terminal_summary(terminalreporter, exitstatus, config):
|
|
|
57
53
|
|
|
58
54
|
# Remove files
|
|
59
55
|
models = [path for x in {"*.onnx", "*.torchscript"} for path in WEIGHTS_DIR.rglob(x)]
|
|
60
|
-
for file in ["decelera_portrait_min.mov", "bus.jpg", "yolo11n.onnx", "yolo11n.torchscript"
|
|
56
|
+
for file in ["decelera_portrait_min.mov", "bus.jpg", "yolo11n.onnx", "yolo11n.torchscript", *models]:
|
|
61
57
|
Path(file).unlink(missing_ok=True)
|
|
62
58
|
|
|
63
59
|
# Remove directories
|
|
64
60
|
models = [path for x in {"*.mlpackage", "*_openvino_model"} for path in WEIGHTS_DIR.rglob(x)]
|
|
65
|
-
for directory in [WEIGHTS_DIR / "path with spaces",
|
|
61
|
+
for directory in [WEIGHTS_DIR / "path with spaces", *models]:
|
|
66
62
|
shutil.rmtree(directory, ignore_errors=True)
|
tests/test_cli.py
CHANGED
|
@@ -8,7 +8,7 @@ from PIL import Image
|
|
|
8
8
|
|
|
9
9
|
from tests import CUDA_DEVICE_COUNT, CUDA_IS_AVAILABLE, MODELS, TASK_MODEL_DATA
|
|
10
10
|
from ultralytics.utils import ARM64, ASSETS, LINUX, WEIGHTS_DIR, checks
|
|
11
|
-
from ultralytics.utils.torch_utils import TORCH_1_11
|
|
11
|
+
from ultralytics.utils.torch_utils import TORCH_1_11, TORCH_2_9, WINDOWS
|
|
12
12
|
|
|
13
13
|
|
|
14
14
|
def run(cmd: str) -> None:
|
|
@@ -82,7 +82,7 @@ def test_fastsam(
|
|
|
82
82
|
everything_results = sam_model(s, device="cpu", retina_masks=True, imgsz=320, conf=0.4, iou=0.9)
|
|
83
83
|
|
|
84
84
|
# Remove small regions
|
|
85
|
-
|
|
85
|
+
_new_masks, _ = Predictor.remove_small_regions(everything_results[0].masks.data, min_area=20)
|
|
86
86
|
|
|
87
87
|
# Run inference with bboxes and points and texts prompt at the same time
|
|
88
88
|
sam_model(source, bboxes=[439, 437, 524, 709], points=[[200, 200]], labels=[1], texts="a photo of a dog")
|
|
@@ -129,3 +129,10 @@ def test_train_gpu(task: str, model: str, data: str) -> None:
|
|
|
129
129
|
def test_solutions(solution: str) -> None:
|
|
130
130
|
"""Test yolo solutions command-line modes."""
|
|
131
131
|
run(f"yolo solutions {solution} verbose=False")
|
|
132
|
+
|
|
133
|
+
|
|
134
|
+
@pytest.mark.skipif(not checks.IS_PYTHON_MINIMUM_3_10 or not TORCH_2_9, reason="Requires Python>=3.10 and Torch>=2.9.0")
|
|
135
|
+
@pytest.mark.skipif(WINDOWS, reason="Skipping test on Windows")
|
|
136
|
+
def test_export_executorch() -> None:
|
|
137
|
+
"""Test exporting a YOLO model to ExecuTorch format via CLI."""
|
|
138
|
+
run("yolo export model=yolo11n.pt format=executorch imgsz=32")
|
tests/test_engine.py
CHANGED
|
@@ -13,7 +13,7 @@ from ultralytics.models.yolo import classify, detect, segment
|
|
|
13
13
|
from ultralytics.utils import ASSETS, DEFAULT_CFG, WEIGHTS_DIR
|
|
14
14
|
|
|
15
15
|
|
|
16
|
-
def test_func(*args):
|
|
16
|
+
def test_func(*args):
|
|
17
17
|
"""Test function callback for evaluating YOLO model performance metrics."""
|
|
18
18
|
print("callback test passed")
|
|
19
19
|
|
tests/test_exports.py
CHANGED
|
@@ -12,15 +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
|
|
16
|
-
|
|
17
|
-
IS_RASPBERRYPI,
|
|
18
|
-
LINUX,
|
|
19
|
-
MACOS,
|
|
20
|
-
WINDOWS,
|
|
21
|
-
checks,
|
|
22
|
-
)
|
|
23
|
-
from ultralytics.utils.torch_utils import TORCH_1_11, TORCH_1_13, TORCH_2_1
|
|
15
|
+
from ultralytics.utils import ARM64, IS_RASPBERRYPI, LINUX, MACOS, WINDOWS, checks
|
|
16
|
+
from ultralytics.utils.torch_utils import TORCH_1_11, TORCH_1_13, TORCH_2_1, TORCH_2_9
|
|
24
17
|
|
|
25
18
|
|
|
26
19
|
def test_export_torchscript():
|
|
@@ -262,3 +255,38 @@ def test_export_imx():
|
|
|
262
255
|
model = YOLO("yolov8n.pt")
|
|
263
256
|
file = model.export(format="imx", imgsz=32)
|
|
264
257
|
YOLO(file)(SOURCE, imgsz=32)
|
|
258
|
+
|
|
259
|
+
|
|
260
|
+
@pytest.mark.skipif(not checks.IS_PYTHON_MINIMUM_3_10 or not TORCH_2_9, reason="Requires Python>=3.10 and Torch>=2.9.0")
|
|
261
|
+
@pytest.mark.skipif(WINDOWS, reason="Skipping test on Windows")
|
|
262
|
+
def test_export_executorch():
|
|
263
|
+
"""Test YOLO model export to ExecuTorch format."""
|
|
264
|
+
file = YOLO(MODEL).export(format="executorch", imgsz=32)
|
|
265
|
+
assert Path(file).exists(), f"ExecuTorch export failed, directory not found: {file}"
|
|
266
|
+
# Check that .pte file exists in the exported directory
|
|
267
|
+
pte_file = Path(file) / Path(MODEL).with_suffix(".pte").name
|
|
268
|
+
assert pte_file.exists(), f"ExecuTorch .pte file not found: {pte_file}"
|
|
269
|
+
# Check that metadata.yaml exists
|
|
270
|
+
metadata_file = Path(file) / "metadata.yaml"
|
|
271
|
+
assert metadata_file.exists(), f"ExecuTorch metadata.yaml not found: {metadata_file}"
|
|
272
|
+
# Note: Inference testing skipped as ExecuTorch requires special runtime setup
|
|
273
|
+
shutil.rmtree(file, ignore_errors=True) # cleanup
|
|
274
|
+
|
|
275
|
+
|
|
276
|
+
@pytest.mark.slow
|
|
277
|
+
@pytest.mark.skipif(not checks.IS_PYTHON_MINIMUM_3_10 or not TORCH_2_9, reason="Requires Python>=3.10 and Torch>=2.9.0")
|
|
278
|
+
@pytest.mark.skipif(WINDOWS, reason="Skipping test on Windows")
|
|
279
|
+
@pytest.mark.parametrize("task", TASKS)
|
|
280
|
+
def test_export_executorch_matrix(task):
|
|
281
|
+
"""Test YOLO export to ExecuTorch format for various task types."""
|
|
282
|
+
file = YOLO(TASK2MODEL[task]).export(format="executorch", imgsz=32)
|
|
283
|
+
assert Path(file).exists(), f"ExecuTorch export failed for task '{task}', directory not found: {file}"
|
|
284
|
+
# Check that .pte file exists in the exported directory
|
|
285
|
+
model_name = Path(TASK2MODEL[task]).with_suffix(".pte").name
|
|
286
|
+
pte_file = Path(file) / model_name
|
|
287
|
+
assert pte_file.exists(), f"ExecuTorch .pte file not found for task '{task}': {pte_file}"
|
|
288
|
+
# Check that metadata.yaml exists
|
|
289
|
+
metadata_file = Path(file) / "metadata.yaml"
|
|
290
|
+
assert metadata_file.exists(), f"ExecuTorch metadata.yaml not found for task '{task}': {metadata_file}"
|
|
291
|
+
# Note: Inference testing skipped as ExecuTorch requires special runtime setup
|
|
292
|
+
shutil.rmtree(file, ignore_errors=True) # cleanup
|
tests/test_integrations.py
CHANGED
|
@@ -8,7 +8,7 @@ from pathlib import Path
|
|
|
8
8
|
|
|
9
9
|
import pytest
|
|
10
10
|
|
|
11
|
-
from tests import MODEL, SOURCE
|
|
11
|
+
from tests import MODEL, SOURCE
|
|
12
12
|
from ultralytics import YOLO, download
|
|
13
13
|
from ultralytics.utils import ASSETS_URL, DATASETS_DIR, SETTINGS
|
|
14
14
|
from ultralytics.utils.checks import check_requirements
|
|
@@ -71,14 +71,14 @@ def test_mlflow_keep_run_active():
|
|
|
71
71
|
|
|
72
72
|
|
|
73
73
|
@pytest.mark.skipif(not check_requirements("tritonclient", install=False), reason="tritonclient[all] not installed")
|
|
74
|
-
def test_triton():
|
|
74
|
+
def test_triton(tmp_path):
|
|
75
75
|
"""Test NVIDIA Triton Server functionalities with YOLO model."""
|
|
76
76
|
check_requirements("tritonclient[all]")
|
|
77
|
-
from tritonclient.http import InferenceServerClient
|
|
77
|
+
from tritonclient.http import InferenceServerClient
|
|
78
78
|
|
|
79
79
|
# Create variables
|
|
80
80
|
model_name = "yolo"
|
|
81
|
-
triton_repo =
|
|
81
|
+
triton_repo = tmp_path / "triton_repo" # Triton repo path
|
|
82
82
|
triton_model = triton_repo / model_name # Triton model path
|
|
83
83
|
|
|
84
84
|
# Export model to ONNX
|