Caffe训练AlexNet网络模型——问题一

训练AlexNet网络时,出现Check failed:datum_height >= crop_size (size vs. 227)错误,具体如下图所示:

根据提示,问题是crop_size的尺寸不匹配,AlexNet网络默认crop_size的尺寸是227*227,而我进行归一化时将每幅图像归一化成了32*32,所以这里出现问题。

在train_val.prototxt文件中将其改为32*32后,上图问题解决,如下图所示:

但紧接着出现下面的问题,如下图所示:

这个问题是由于归一化后的尺寸太小了,使AlexNet网络训练到第五层卷积层就已经使batch size = 1了,这是由于AlexNet网络结构有八层,所以图像的尺寸大小最好在256*256以上,并且不要修改AlexNet网络默认crop_size的尺寸,使其保持是227*227。

原文地址:https://www.cnblogs.com/hust-yingjie/p/6526370.html