caffe训练数据时出现了Check failed: data_怎么办?

来源:互联网 发布:优化营商环境总结报告 编辑:程序博客网 时间:2024/05/16 10:41

如下图所示:



原因:

训练样本的图像尺寸太小,在池化层pool5的时候,输入图像的尺寸已经小于kernel核的大小了,经过池化之后,下一步输入就变成了0x0,因此会报错。

解决办法:

在转换图像格式的时候,设置宽度和高度即可。

convert_imageset.exe --backend=leveldb --resize_width=64 --resize_height=64

我这边直接调整到64*64就可以了。




阅读全文
0 0