如下图所示:
原因:
训练样本的图像尺寸太小,在池化层pool5的时候,输入图像的尺寸已经小于kernel核的大小了,经过池化之后,下一步输入就变成了0x0,因此会报错。
解决办法:
在转换图像格式的时候,设置宽度和高度即可。
convert_imageset.exe --backend=leveldb --resize_width=64 --resize_height=64
我这边直接调整到64*64就可以了。
如下图所示:
原因:
训练样本的图像尺寸太小,在池化层pool5的时候,输入图像的尺寸已经小于kernel核的大小了,经过池化之后,下一步输入就变成了0x0,因此会报错。
解决办法:
在转换图像格式的时候,设置宽度和高度即可。
convert_imageset.exe --backend=leveldb --resize_width=64 --resize_height=64
我这边直接调整到64*64就可以了。