-
Notifications
You must be signed in to change notification settings - Fork 1
/
p.prototxt
113 lines (109 loc) · 1.43 KB
/
p.prototxt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
name: "pNet"
layer {
name: "data"
type: "Input"
top: "data"
input_param {
shape: {
dim: 1
dim: 3
dim: 12
dim: 12
}
}
}
layer {
name: "conv1"
type: "Convolution"
bottom: "data"
top: "conv1"
convolution_param {
num_output: 10
kernel_size: 3
stride: 1
}
}
layer {
name: "prelu1"
type: "PReLU"
bottom: "conv1"
top: "conv1"
}
layer {
name: "pool1"
type: "Pooling"
bottom: "conv1"
top: "pool1"
pooling_param {
pool: MAX
kernel_size: 2
stride: 2
}
}
layer {
name: "conv2"
type: "Convolution"
bottom: "pool1"
top: "conv2"
convolution_param {
num_output: 16
kernel_size: 3
stride: 1
}
}
layer {
name: "prelu2"
type: "PReLU"
bottom: "conv2"
top: "conv2"
}
layer {
name: "conv3"
type: "Convolution"
bottom: "conv2"
top: "conv3"
convolution_param {
num_output: 32
kernel_size: 3
stride: 1
}
}
layer {
name: "prelu3"
type: "PReLU"
bottom: "conv3"
top: "conv3"
}
# score
layer {
name: "score"
type: "Convolution"
bottom: "conv3"
top: "score"
convolution_param {
num_output: 2
kernel_size: 1
stride: 1
}
}
layer {
name: "prob"
type: "Softmax"
bottom: "score"
top: "prob"
softmax_param {
axis: 1
}
}
# bbox
layer {
name: "bbox_pred"
type: "Convolution"
bottom: "conv3"
top: "bbox_pred"
convolution_param {
num_output: 4
kernel_size: 1
stride: 1
}
}