dgenerate-ultralytics-headless 8.3.220__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.220.dist-info → dgenerate_ultralytics_headless-8.3.221.dist-info}/METADATA +1 -1
- {dgenerate_ultralytics_headless-8.3.220.dist-info → dgenerate_ultralytics_headless-8.3.221.dist-info}/RECORD +73 -73
- tests/__init__.py +5 -7
- tests/conftest.py +3 -7
- tests/test_cli.py +1 -1
- tests/test_engine.py +1 -1
- 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/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 +16 -16
- ultralytics/engine/model.py +1 -1
- 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/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 +5 -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 +1 -1
- 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 +4 -4
- ultralytics/utils/triton.py +2 -2
- ultralytics/utils/tuner.py +4 -2
- {dgenerate_ultralytics_headless-8.3.220.dist-info → dgenerate_ultralytics_headless-8.3.221.dist-info}/WHEEL +0 -0
- {dgenerate_ultralytics_headless-8.3.220.dist-info → dgenerate_ultralytics_headless-8.3.221.dist-info}/entry_points.txt +0 -0
- {dgenerate_ultralytics_headless-8.3.220.dist-info → dgenerate_ultralytics_headless-8.3.221.dist-info}/licenses/LICENSE +0 -0
- {dgenerate_ultralytics_headless-8.3.220.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,18 +1,18 @@
|
|
|
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=
|
|
6
|
+
tests/test_engine.py,sha256=ER2DsHM0GfUG99AH1Q-Lpm4x36qxkfOzxmH6uYM75ds,5722
|
|
7
7
|
tests/test_exports.py,sha256=OMLio2uUhyqo8D8qB5xUwmk7Po2rMeAACRc8WYoxbj4,13147
|
|
8
|
-
tests/test_integrations.py,sha256=
|
|
9
|
-
tests/test_python.py,sha256=
|
|
10
|
-
tests/test_solutions.py,sha256=
|
|
11
|
-
ultralytics/__init__.py,sha256=
|
|
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=
|
|
15
|
+
ultralytics/cfg/__init__.py,sha256=jqOp7PCEN7vc-ItBiPFne3k8jagkXnL_Ep-IryuyhxA,40017
|
|
16
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
|
|
@@ -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=
|
|
125
|
+
ultralytics/engine/exporter.py,sha256=dFbnKY_6Ym0aY6gk1GYspgGTHI6UA6LSLUJaxAAwPDU,73543
|
|
126
|
+
ultralytics/engine/model.py,sha256=sCa5W6Qit5NloEVuuvNelH-geJQw9IGjcOc9_PXNDnI,53456
|
|
127
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
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
|
@@ -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")
|
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_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
|