溫馨提示×
您好，登錄后才能下訂單哦！
點擊 登錄注冊 即表示同意《億速云用戶服務條款》
您好，登錄后才能下訂單哦！
C++ 將文件數據一次性加載進內存實例代碼
問題：
早先寫了一個目標檢測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<long>((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<int> 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) 然后調用從內存加載的接口；
感謝閱讀，希望能幫助到大家，謝謝大家對本站的支持！
免責聲明：本站發布的內容（圖片、視頻和文字）以原創、轉載和分享為主，文章觀點不代表本網站立場，如果涉及侵權請聯系站長郵箱：is@yisu.com進行舉報，并提供相關證據，一經查實，將立刻刪除涉嫌侵權內容。
