所以我的问题在于无法使用较小的图像(32X32)训练imagenet,当我将它们调整为256X256时,一切都开始训练。所以我知道问题是我的设置。
我试图设置我自己的设置:
deploy.prototxt:我将最后2个input_dims设置为32
solver.prototxt:我设置了solver_mode:CPU(单独留下其他所有东西)
train_val.prototxt:我为这两个设置设置了crop_size:31
所有路径都正确,因为训练运行正常,图像大小调整为256X256
但是使用上面描述的设置我得到了这个错误:
I1114 11:53:38.484948 4566 net.cpp:96] Setting up fc6
F1114 11:53:38.484967 4566 blob.cpp:72] Check failed: data_
*** Check failure stack trace: ***
@ 0x7f82ed8a0daa (unknown)
@ 0x7f82ed8a0ce4 (unknown)
@ 0x7f82ed8a06e6 (unknown)
@ 0x7f82ed8a3687 (unknown)
@ 0x45814e caffe::Blob<>::mutable_cpu_data()
@ 0x4c12fa caffe::GaussianFiller<>::Fill()
@ 0x4e5719 caffe::InnerProductLayer<>::LayerSetUp()
@ 0x4705e1 caffe::Net<>::Init()
@ 0x471eee caffe::Net<>::Net()
@ 0x452560 caffe::Solver<>::InitTrainNet()
@ 0x4537c6 caffe::Solver<>::Init()
@ 0x453926 caffe::Solver<>::Solver()
@ 0x415650 caffe::GetSolver<>()
@ 0x411b44 train()
@ 0x40ef21 main
@ 0x7f82ea913ec5 (unknown)
@ 0x410697 (unknown)
@ (nil) (unknown)
train_caffenet_letter.sh: line 4: 4566 Aborted (core dumped) ./build/tools/caffe train --solver=models/bvlc_reference_caffenet/solver_letter.prototxt
如果有人可以解释我应该在这些文件中更改哪些其他设置,或者我完全错了?
我试过按照这个教程但是没有放弃它: http://drubiano.github.io/2014/06/18/caffe-custom-data.html