C++将文件数据一次性加载进内存实例代码-成都创新互联网站建设

关于创新互联

多方位宣传企业产品与服务 突出企业形象

公司简介 公司的服务 荣誉资质 新闻动态 联系我们

C++将文件数据一次性加载进内存实例代码

C++ 将文件数据一次性加载进内存实例代码

象山网站制作公司哪家好,找创新互联!从网页设计、网站建设、微信开发、APP开发、响应式网站建设等网站项目制作,到程序开发,运营维护。创新互联成立于2013年到现在10年的时间,我们拥有了丰富的建站经验和运维经验,来保证我们的工作的顺利进行。专注于网站建设就选创新互联

问题:

早先写了一个目标检测SDK,里面有从bin文件加载模型和从内存加载模型两个接口。后来遇到了级联检测,即有多个bin模型文件,当想要把多个bin文件合并成一个的时候,发现对应的加载接口也得变。

解决:

为了不改变接口,采用了下面的解决思路:

(1) 将多个bin文件进行拼接,同时记录每个文件的大小.

合并后的文件为: 模型文件个数+模型A大小+…+模型X大小+模型A参数…

(2) 采用下面的方法将这个合并的文件一次性加载进内存

 /**************** 将模型整个读入内存  ********************/
  std::ifstream infile(detModFile,std::ios::binary);
  if (!infile.is_open()) 
  { 
    printf( "connot open the model file: %s\n",detModFile); 
    return -1; 
  } 
  std::filebuf *pbuf = infile.rdbuf(); 
  // 获取文件大小
  long filesize = static_cast((pbuf->pubseekoff (0,std::ios::end,std::ios::in))); 
  pbuf->pubseekpos (0,std::ios::in);
  unsigned char* modelptr = new unsigned char[filesize]; 
  // 读入文件内容
  pbuf->sgetn ((char*)modelptr,filesize); 
  infile.close();   

  // 获取模型个数
  int model_num;
  int p_offset = 0;
  memcpy(&model_num,modelptr,sizeof(int));
  p_offset += sizeof(int);

  //获取每个模型大小
  std::vector each_size(model_num);
  int model_size;
  for (int i = 0; i < model_num; i++)
  {
    memcpy(&model_size,modelptr+p_offset,sizeof(int));
    p_offset += sizeof(int);
    each_size[i] = model_size;
  }

(3) 然后调用从内存加载的接口;

感谢阅读,希望能帮助到大家,谢谢大家对本站的支持!


名称栏目:C++将文件数据一次性加载进内存实例代码
标题链接:http://kswsj.cn/article/jgdsgo.html

其他资讯