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

时间:2016-06-23 13:41:08

标签: caffe

我正在使用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)

1 个答案:

答案 0 :(得分:0)

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