Skip to content

Commit fcf83be

Browse files
committed
add link to download vit_b_32
1 parent 10702ac commit fcf83be

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

testdata/dnn/download_models.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1131,6 +1131,13 @@ 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+
],
1139+
sha='88144dca52cf3c6fee3aed8f8ca5c0b431e0afbd',
1140+
filename='onnx/models/vit_b_32.onnx'),
11341141
]
11351142

11361143
# Note: models will be downloaded to current working directory

0 commit comments

Comments
 (0)