您的位置:首页 > Web前端

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

2017-03-09 16:33 253 查看
训练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。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐