Skip to content

Commit b94ede6

Browse files
committed
add model and data
1 parent 96169f0 commit b94ede6

12 files changed

+259
-85
lines changed

testdata/dnn/download_models.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1131,6 +1131,21 @@ def is_archive(self):
11311131
url='https://github.com/opencv/opencv_zoo/raw/281d232cd99cd920853106d853c440edd35eb442/models/optical_flow_estimation_raft/optical_flow_estimation_raft_2023aug.onnx',
11321132
sha='8165e43c7bd37cc651f66509532acdb3c292832b',
11331133
filename='onnx/models/optical_flow_estimation_raft_2023aug.onnx'),
1134+
Model(
1135+
name='vit_b_32',
1136+
url=[
1137+
'https://drive.google.com/u/0/uc?id=1UEeAyBs76XVkypk56ou7B8rBEIAlirkD&export=download', # See https://github.com/opencv/opencv_extra/pull/1128 to generate this model from torchvision
1138+
'https://dl.opencv.org/models/vit/vit_b_32.onnx',
1139+
],
1140+
sha='88144dca52cf3c6fee3aed8f8ca5c0b431e0afbd',
1141+
filename='onnx/models/vit_b_32.onnx'),
1142+
Model(
1143+
name='object_tracking_vittrack',
1144+
url=[
1145+
'https://github.com/opencv/opencv_zoo/raw/fef72f8fa7c52eaf116d3df358d24e6e959ada0e/models/object_tracking_vittrack/object_tracking_vittrack_2023sep.onnx',
1146+
],
1147+
sha='50008bb4f6a27b1aa940ad886b1bd1936ac4ed3e',
1148+
filename='onnx/models/object_tracking_vittrack_2023sep.onnx'),
11341149
]
11351150

11361151
# Note: models will be downloaded to current working directory
60.1 KB
Binary file not shown.
60.1 KB
Binary file not shown.
60.1 KB
Binary file not shown.
60.1 KB
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
4.03 KB
Binary file not shown.

testdata/dnn/onnx/generate_onnx_models_with_onnxscript.py

Lines changed: 244 additions & 85 deletions
Large diffs are not rendered by default.

0 commit comments

Comments
 (0)