name: "CBCL"
layers {
name: "data"
type: IMAGE_DATA
top: "data"
top: "label"
image_data_param {
source: "/tmp/train.txt"
batch_size: 10
shuffle: true
}
include: { phase: TRAIN }
}
layers {
name: "data"
type: IMAGE_DATA
top: "data"
top: "label"
image_data_param {
source: "/tmp/test.txt"
batch_size: 10
}
include: { phase: TEST }
}
layers {
name: "C1"
type: CONVOLUTION
top: "C1"
bottom: "data"
convolution_param {
kernel_size: 3
num_output: 8
}
}
layers {
name: "C2"
type: CONVOLUTION
top: "C2"
bottom: "C1"
convolution_param {
kernel_size: 4
num_output: 32
}
}
layers {
name: "S1"
type: POOLING
top: "S1"
bottom: "C2"
pooling_param {
pool: MAX
kernel_size: 2
}
}
layers {
name: "C3"
type: CONVOLUTION
top: "C3"
bottom: "S1"
convolution_param {
kernel_size: 4
num_output: 128
}
}
layers {
name: "C4"
type: CONVOLUTION
top: "C4"
bottom: "C3"
convolution_param {
kernel_size: 4
num_output: 1024
}
}
layers {
name: "F1"
top: "F1"
bottom: "C4"
type: INNER_PRODUCT
base_lr: 0.01
lr_policy: "step"
gamma: 0.1
stepsize: 100000
max_iter: 350000
momentum: 0.9
inner_product_param {
num_output: 2
}
}
layers {
name: "loss"
type: SOFTMAX_LOSS
bottom: "F1"
bottom: "label"
top: "loss"
}