caffe - 如何减少用于 googlenet 训练的输入图像的大小

Question

我正在使用 googlenet 模型，并且我想使用不同的图像大小进行训练（图像的较小版本），image_mean 和 lmdb 文件的大小也减小了。当我修改 googlenet.prototxt 中的大小并运行 train.sh 文件时，程序被中止并出现以下错误。

• 检查失败堆栈跟踪：*** @ 0x7fdbb9499daa（未知） @ 0x7fdbb9499ce4（未知） @ 0x7fdbb94996e6（未知） @ 0x7fdbb949c687（未知） @ 0x7fdbb99543ee caffe::Blob::Reshape() @ 0x7fdbb9954619 caffe::Blob::Reshape() @ 0x7fdbb98dcc08 caffe::PoolingLayer::Reshape() @ 0x7fdbb99968ac caffe::Net::Init() @ 0x7fdbb9997725 caffe::Net::Net() @ 0x7fdbb985174a caffe::Solver::InitTrainNet() @ 0x7fdbb985284c caffe::Solver::Init() @ 0x7fdbb9852b7a caffe::Solver::Solver() @ 0x7fdbb984a263 caffe::Creator_SGDSolver() @ 0x40e67e caffe::SolverRegistry::CreateSolver() @ 0x40794b 火车（） @ 0x40590c 主 @ 0x7fdbb87a7f45（未知） @ 0x406041（未知） @ (nil) (unknown) Aborted (core dumped)

Answer 1

我发现googlenet模型的大小是固定的并且是唯一的，所以修改它会导致错误。因此，必须保持与模型训练所需的大小相同。