读取caffe数据集文件txt

最新推荐文章于 2023-10-25 16:30:47 发布

qulay

最新推荐文章于 2023-10-25 16:30:47 发布

阅读量1.3k

点赞数

分类专栏： C/C++

本文链接：https://blog.csdn.net/klqulei123/article/details/52781395

版权

本文介绍如何读取Caffe用于训练的数据集TXT文件，这些文件包含每个样本的文件路径及其对应的类别标签信息。

摘要由CSDN通过智能技术生成

读取caffe训练数据文件txt，每行包括文件地址与类别标签。

  string src;
  std::ifstream infile(src);
  std::vector<std::pair<std::string, int> > lines;
  std::string filename;
  int label;
  while (infile >> filename >> label) {
    lines.push_back(std::make_pair(filename, label));
  }

 for (int line_id = 0; line_id < lines.size(); ++line_id) {
      string fn = lines[line_id].first;
      size_t p = fn.rfind('.');
      std::string enc = fn.substr(p);
      if ( en == "png" )
      {
          std::vector<uchar> buf;
          cv::imencode("."+enc , cv_img, buf);
      }
    
}

bool ReadFileToDatum(const string& filename, const int label,Datum* datum) 
{
	std::streampos size;
	fstream file(filename.c_str(), ios::in | ios::binary | ios::ate);
	if