Skip to content

Commit 577348c

Browse files
committed
fixup! Refactored download_models.py - split functionality - support multiple URLs - reduce output - support new yolo models
1 parent 3ec46bb commit 577348c

File tree

1 file changed

+11
-2
lines changed

1 file changed

+11
-2
lines changed

testdata/dnn/download_models.py

Lines changed: 11 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1002,8 +1002,17 @@ def is_archive(self):
10021002
'https://dl.opencv.org/models/yolox/yolox_s_inf_decoder.onnx'
10031003
],
10041004
sha='b205b00122cc7bf559a0e845680408320df3a898',
1005-
filename='onnx/models/yolox_s_inf_decoder.onnx'
1006-
),
1005+
filename='onnx/models/yolox_s_inf_decoder.onnx'),
1006+
Model(
1007+
name='yolov6n',
1008+
url='https://dl.opencv.org/models/yolov6/yolov6n.onnx',
1009+
sha='a704c0ace51103a43920c50a396b2c8b09d2daec',
1010+
filename='onnx/models/yolov6n.onnx'),
1011+
Model(
1012+
name='yolov8n',
1013+
url='https://dl.opencv.org/models/yolov8/yolov8n.onnx',
1014+
sha='136807b88d0b02bc226bdeb9741141d857752e10',
1015+
filename='onnx/models/yolov8n.onnx'),
10071016
Model(
10081017
name='NanoTrackV1 (ONNX, backbone)',
10091018
url='https://raw.githubusercontent.com/zihaomu/opencv_extra_data_backup/main/NanoTrack/models/nanotrack_backbone_sim.onnx',

0 commit comments

Comments
 (0)