Skip to content

Commit 80ae3bb

Browse files
committed
Merge remote-tracking branch 'upstream/3.4' into merge-3.4
2 parents bf5df9c + e5de61c commit 80ae3bb

17 files changed

+1392
-359
lines changed
-234 Bytes
Loading
-2.16 KB
Loading
-1.91 KB
Loading
-2.02 KB
Loading
-2 KB
Loading

testdata/dnn/download_models.py

Lines changed: 2 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -110,8 +110,8 @@ def save(self, r):
110110
sha='c8a5a000ee8d8dd75886f152a50a9c5b53d726a5',
111111
filename='tensorflow_inception_graph.pb'),
112112
Model(
113-
name='Enet',
114-
url='https://www.dropbox.com/sh/dywzk3gyb12hpe5/AABoUwqQGWvClUu27Z1EWeu9a/model-best.net?dl=1',
113+
name='Enet', # https://github.com/e-lab/ENet-training
114+
url='https://www.dropbox.com/s/tdde0mawbi5dugq/Enet-model-best.net?dl=1',
115115
sha='b4123a73bf464b9ebe9cfc4ab9c2d5c72b161315',
116116
filename='Enet-model-best.net'),
117117
Model(
@@ -437,29 +437,6 @@ def save(self, r):
437437
member='zfnet512/test_data_set_0/output_0.pb',
438438
sha='a74974096088954ca4e4e89bec212c1ac2ab0745',
439439
filename='onnx/data/output_zfnet512.pb'),
440-
Model(
441-
name='VGG16 (ONNX)',
442-
url='https://s3.amazonaws.com/onnx-model-zoo/vgg/vgg16/vgg16.tar.gz',
443-
sha='39bc0f689388a32227d804e97ea9f562467ed863',
444-
filename='vgg16.tar.gz'),
445-
Model(
446-
name='VGG16 (ONNX)',
447-
archive='vgg16.tar.gz',
448-
member='vgg16/vgg16.onnx',
449-
sha='effed6f2bbe6c3a2f7705da4c0002ec6250827dd',
450-
filename='onnx/models/vgg16.onnx'),
451-
Model(
452-
name='VGG16 (ONNX)',
453-
archive='vgg16.tar.gz',
454-
member='vgg16/test_data_set_0/input_0.pb',
455-
sha='55c285cfbc4d61e3c026302a3af9e7d220b82d0a',
456-
filename='onnx/data/input_vgg16.pb'),
457-
Model(
458-
name='VGG16 (ONNX)',
459-
archive='vgg16.tar.gz',
460-
member='vgg16/test_data_set_0/output_0.pb',
461-
sha='dcd5945025906ff4ea2270963b795c8fa29d4cd8',
462-
filename='onnx/data/output_vgg16.pb'),
463440
Model(
464441
name='VGG16_bn (ONNX)',
465442
url='https://s3.amazonaws.com/onnx-model-zoo/vgg/vgg16-bn/vgg16-bn.tar.gz',
Lines changed: 37 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,37 @@
1+
name: "prior_box"
2+
input: "data"
3+
input_shape {
4+
dim: 1
5+
dim: 3
6+
dim: 10
7+
dim: 10
8+
}
9+
10+
input: "shape"
11+
input_shape {
12+
dim: 1
13+
dim: 2
14+
dim: 3
15+
dim: 4
16+
}
17+
18+
layer {
19+
name: "priorbox"
20+
type: "PriorBox"
21+
bottom: "shape"
22+
bottom: "data"
23+
top: "priorbox"
24+
prior_box_param {
25+
min_size: 2.0
26+
min_size: 3.0
27+
max_size: 6.0
28+
max_size: 7.0
29+
aspect_ratio: 2
30+
flip: true
31+
clip: false
32+
variance: 0.1
33+
variance: 0.1
34+
variance: 0.2
35+
variance: 0.2
36+
}
37+
}
3.13 KB
Binary file not shown.
224 Bytes
Binary file not shown.
128 Bytes
Binary file not shown.

testdata/dnn/onnx/generate_onnx_models.py

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -425,3 +425,17 @@ def forward(self, x):
425425
input = np.random.rand(1, 3, 4, 2, 3)
426426
output = np.mean(input, axis=(3, 4), keepdims=True)
427427
save_onnx_data_and_model(input, output, 'reduce_mean3d', 'ReduceMean', axes=(3, 4), keepdims=True)
428+
429+
class SplitMax(nn.Module):
430+
431+
def __init__(self):
432+
super(SplitMax, self).__init__()
433+
434+
def forward(self, x):
435+
first, second = torch.split(x, (2, 4), dim=1)
436+
second, third = torch.split(second, (2, 2), dim=1)
437+
return torch.max(first, torch.max(second, third))
438+
439+
model = SplitMax()
440+
input = Variable(torch.randn(1, 6, 2, 3))
441+
save_data_and_model("split_max", input, model)
Lines changed: 27 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,27 @@
1+
pytorch1.1:�
2+
9
3+
tensor.11tensor"Split*
4+
axis�*
5+
split@@�
6+
2
7+
tensor34"Split*
8+
axis�*
9+
split@@�
10+

11+
3
12+
45"Max
13+

14+
1
15+
56"Maxtorch-jit-exportZ"
16+
tensor.1
17+

18+

19+

20+

21+
b
22+
6
23+

24+

25+

26+

27+
B

testdata/dnn/opencv_face_detector.pbtxt

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1221,7 +1221,7 @@ node {
12211221
attr {
12221222
key: "padding"
12231223
value {
1224-
s: "SAME"
1224+
s: "VALID"
12251225
}
12261226
}
12271227
attr {
@@ -1311,7 +1311,7 @@ node {
13111311
attr {
13121312
key: "padding"
13131313
value {
1314-
s: "SAME"
1314+
s: "VALID"
13151315
}
13161316
}
13171317
attr {
@@ -2337,6 +2337,12 @@ node {
23372337
i: 400
23382338
}
23392339
}
2340+
attr {
2341+
key: "clip"
2342+
value {
2343+
b: true
2344+
}
2345+
}
23402346
}
23412347
node {
23422348
name: "reshape_before_softmax"

testdata/dnn/opencv_face_detector.prototxt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1785,5 +1785,6 @@ layer {
17851785
code_type: CENTER_SIZE
17861786
keep_top_k: 200
17871787
confidence_threshold: 0.01
1788+
clip: 1
17881789
}
17891790
}

0 commit comments

Comments
 (0)