C++ 将文件数据一次性加载进内存

最新推荐文章于 2023-04-26 16:17:37 发布

原创

最新推荐文章于 2023-04-26 16:17:37 发布 · 5.4k 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#C++

本文介绍了如何在C++中将多个合并的bin模型文件一次性加载到内存，以应对级联检测场景。通过拼接模型文件并记录每个文件的大小，然后在内存中加载整个合并文件，保持接口不变。

问题： 早先写了一个目标检测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