您的位置:首页 > Web前端

Caffe在转化图片到levelDB中遇到问题Check failed: data.size() == data_size

2014-07-14 21:13 399 查看
在把图片转化到levelDB中遇到了Check failed: data.size() == data_size,归根究底还是源码没细看,找到出错的行在F0714 20:31:14.899121 26565 convert_imageset.cpp:84] convert_imageset.cpp中的第84行, CHECK_EQ(data.size(), data_size) << "Incorrect data field size " << data.size();就是说两个大小不一致,再看代码
int data_size;

bool data_size_initialized = false;

for (int line_id = 0; line_id < lines.size(); ++line_id) {

if (!ReadImageToDatum(root_folder + lines[line_id].first,lines[line_id].second, datum)) {

continue;

}

if (!data_size_initialized) {

data_size = datum.channels() * datum.height() * datum.width();

data_size_initialized = true;

} else {

const string& data = datum.data();

CHECK_EQ(data.size(), data_size) << "Incorrect data field size "

<< data.size();

}

从上面的代码可知,第一次循环中,data_size_initialized=false,然后进入到if (!data_size_initialized) 中,把data_size设为了datum.channels() * datum.height() * datum.width(),同时把data_size_initialized=true,在以后的迭代中,都是执行else语句,从而加入图片大小不一致会报错,处理的办法可选的是,在转换到数据库levelDB前,让图片resize到一样的大小,或者把ReadImageToDatum改成ReadImageToDatum(root_folder
+ lines[line_id].first,lines[line_id].second,width,height ,datum)。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: