【发布时间】:2016-05-29 03:45:22
【问题描述】:
现在,我是具有 2 类数据的训练网络……但在第一次迭代后准确度是恒定的 1!
输入数据是灰度图像。创建 HDF5Data 时,两个类图像都是随机选择的。
为什么会这样?哪里错了或者哪里错了!
network.prototxt:
name: "brainMRI"
layer {
name: "data"
type: "HDF5Data"
top: "data"
top: "label"
include: {
phase: TRAIN
}
hdf5_data_param {
source: "/home/shivangpatel/caffe/brainMRI1/train_file_location.txt"
batch_size: 10
}
}
layer {
name: "data"
type: "HDF5Data"
top: "data"
top: "label"
include: {
phase: TEST
}
hdf5_data_param {
source: "/home/shivangpatel/caffe/brainMRI1/test_file_location.txt"
batch_size: 10
}
}
layer {
name: "conv1"
type: "Convolution"
bottom: "data"
top: "conv1"
param {
lr_mult: 1
}
param {
lr_mult: 2
}
convolution_param {
num_output: 20
kernel_size: 5
stride: 1
weight_filler {
type: "xavier"
}
bias_filler {
type: "constant"
}
}
}
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"
param {
lr_mult: 1
}
param {
lr_mult: 2
}
convolution_param {
num_output: 50
kernel_size: 5
stride: 1
weight_filler {
type: "xavier"
}
bias_filler {
type: "constant"
}
}
}
layer {
name: "pool2"
type: "Pooling"
bottom: "conv2"
top: "pool2"
pooling_param {
pool: MAX
kernel_size: 2
stride: 2
}
}
layer {
name: "ip1"
type: "InnerProduct"
bottom: "pool2"
top: "ip1"
param {
lr_mult: 1
}
param {
lr_mult: 2
}
inner_product_param {
num_output: 500
weight_filler {
type: "xavier"
}
bias_filler {
type: "constant"
}
}
}
layer {
name: "relu1"
type: "ReLU"
bottom: "ip1"
top: "ip1"
}
layer {
name: "ip2"
type: "InnerProduct"
bottom: "ip1"
top: "ip2"
param {
lr_mult: 1
}
param {
lr_mult: 2
}
inner_product_param {
num_output: 2
weight_filler {
type: "xavier"
}
bias_filler {
type: "constant"
}
}
}
layer {
name: "softmax"
type: "Softmax"
bottom: "ip2"
top: "smip2"
}
layer {
name: "loss"
type: "SoftmaxWithLoss"
bottom: "ip2"
bottom: "label"
top: "loss"
}
layer {
name: "accuracy"
type: "Accuracy"
bottom: "smip2"
bottom: "label"
top: "accuracy"
include {
phase: TEST
}
}
输出:
I0217 17:41:07.912580 2913 net.cpp:270] This network produces output loss I0217 17:41:07.912607 2913 net.cpp:283] Network initialization done. I0217 17:41:07.912739 2913 solver.cpp:60] Solver scaffolding done. I0217 17:41:07.912789 2913 caffe.cpp:212] Starting Optimization I0217 17:41:07.912813 2913 solver.cpp:288] Solving brainMRI I0217 17:41:07.912832 2913 solver.cpp:289] Learning Rate Policy: inv I0217 17:41:07.920737 2913 solver.cpp:341] Iteration 0, Testing net (#0) I0217 17:41:08.235076 2913 solver.cpp:409] Test net output #0: accuracy = 0.98 I0217 17:41:08.235194 2913 solver.cpp:409] Test net output #1: loss = 0.0560832 (* 1 = 0.0560832 loss) I0217 17:41:35.831647 2913 solver.cpp:341] Iteration 100, Testing net (#0) I0217 17:41:36.140849 2913 solver.cpp:409] Test net output #0: accuracy = 1 I0217 17:41:36.140949 2913 solver.cpp:409] Test net output #1: loss = 0.00757247 (* 1 = 0.00757247 loss) I0217 17:42:05.465395 2913 solver.cpp:341] Iteration 200, Testing net (#0) I0217 17:42:05.775877 2913 solver.cpp:409] Test net output #0: accuracy = 1 I0217 17:42:05.776000 2913 solver.cpp:409] Test net output #1: loss = 0.0144996 (* 1 = 0.0144996 loss) ............. .............
【问题讨论】:
-
你的测试集大小是多少?
-
h5ls test_AD.hdf5
data Dataset {3686, 1, 45, 64} label Dataset {3686, 1, 1, 1}h5ls test_NC.hdf5data Dataset {7334, 1, 45, 64} label Dataset {7334, 1, 1, 1}h5ls train_AD.hdf5data Dataset {14746, 1, 45, 64} label Dataset {14746, 1, 1, 1}h5ls train_NC.hdf5data Dataset {29338, 1, 45, 64} label Dataset {29338, 1, 1, 1} -
您的日志仅打印测试迭代,您是否在进行任何培训?您在每个测试间隔运行多少次测试迭代?
-
test_iter: 5 test_interval: 100 base_lr: 0.001 momentum: 0.9 weight_decay: 0.0005 lr_policy: "inv" gamma: 0.0001 power: 0.75 # The maximum number of iterations max_iter: 10000 # snapshot intermediate results snapshot: 1000 snapshot_prefix: "/home/shivangpatel/caffe/brainMRI1/" # solver mode: CPU or GPU solver_mode: CPU -
有关
test_interval和batch_size的更多信息,请参见this answer。
标签: machine-learning neural-network deep-learning caffe training-data