Skip to content

Commit 800a8c4

Browse files
authored
Merge pull request #1128 from fengyuentau:attention_layer
dnn: add model and data for attention layer implementation
2 parents 96169f0 + b94ede6 commit 800a8c4

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)