Skip to content

Commit 25bc1f7

Browse files
Fix prototxt for Alexnet, FaceDetect-V1 and GoogleNet
1 parent c782046 commit 25bc1f7

File tree

3 files changed

+6
-4
lines changed

3 files changed

+6
-4
lines changed

caffe_models/alexnet/caffe_model/deploy.prototxt

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,8 @@ name: "AlexNet"
77
#}
88
input : "data"
99
input_shape {
10-
dim: 10
10+
# dim: 10 # No need for batch=10 in deployment
11+
dim: 1
1112
dim: 3
1213
dim: 227
1314
dim: 227

caffe_models/facedetect_v1/caffe_model/face-deploytrain.prototxt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
name: "face-mcw-110"
22
input: "data"
3-
input_dim: 128
3+
input_dim: 1
44
input_dim: 1
55
input_dim: 96
66
input_dim: 96
@@ -159,4 +159,4 @@ layer {
159159
type: "Softmax"
160160
bottom: "ip1"
161161
top: "prob"
162-
}
162+
}

caffe_models/googlenet/caffe_model/deploy.prototxt

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,8 @@ layer {
33
name: "data"
44
type: "Input"
55
top: "data"
6-
input_param { shape: { dim: 10 dim: 3 dim: 224 dim: 224 } }
6+
# input_param { shape: { dim: 10 dim: 3 dim: 224 dim: 224 } }
7+
input_param { shape: { dim: 1 dim: 3 dim: 224 dim: 224 } }
78
}
89
layer {
910
name: "conv1/7x7_s2"

0 commit comments

Comments
 (0)