C++ 將文件數據一次性加載進內存實例代碼。本站提示廣大學習愛好者:(C++ 將文件數據一次性加載進內存實例代碼)文章只能為提供參考,不一定能成為您想要的結果。以下是C++ 將文件數據一次性加載進內存實例代碼正文
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) 然後調用從內存加載的接口;
感謝閱讀,希望能幫助到大家,謝謝大家對本站的支持!