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

最新推荐文章于 2022-05-12 14:11:22 发布

Wanglan_Alan

最新推荐文章于 2022-05-12 14:11:22 发布

阅读量9.2k

点赞数

文章标签： Caffe

本文链接：https://blog.youkuaiyun.com/Alan317/article/details/37772457

版权

文章详细介绍了在将图片转化为LevelDB过程中遇到的数据大小不一致问题，通过分析代码发现原因在于图片大小不一致导致的错误。提出了两种解决方案：在数据库插入前统一图片大小，或修改读取图片函数以接受指定宽度和高度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在把图片转化到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)。