扫二维码与项目经理沟通
我们在微信上24小时期待你的声音
解答本文疑问/技术咨询/运营咨询/技术建议/互联网交流
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) 然后调用从内存加载的接口；
感谢阅读，希望能帮助到大家，谢谢大家对本站的支持！
我们在微信上24小时期待你的声音
解答本文疑问/技术咨询/运营咨询/技术建议/互联网交流
微信二维码
移动版官网