主要是今天工作的時候需要把一些數據保存到本地,因為是一些預測值和標簽的對比,還有預測值的概率,所以想到用CSV文件來保存,大概查了一下,還是比較簡單的,所以記錄一下。
首先要說明的是CSV文件有點類似excel文件,也可以用excel文件直接打開的。在寫數據到CSV文件的時候要注意,字符串逗號表示的是換列,換行符號就是換行,所以在寫數據的時候要注意好這兩點就可以了。
然后寫的時候也很簡單,直接用C++的ofstream就可以實現了,ofstream有一個操作是"<<",這個也很好用,就類似std::cout的操作一樣即可。比如我現在的數據是比較統一的,每一個樣本是一行,一行數據要分成四列,第一列是樣本的圖像地址,第二列是標簽,第三列是最終預測值,第四列是概率,每一列的格式是一樣的,那么我的代碼就是這樣:
ofstream file(CSV_PATH);
if (file)
{
file << image_path << "," << label << "," << prediction << "," << probability << "\n";
}
file.close();
這樣就可以實現了,還是比較容易的。
雲 灰灰的
再也洗不干凈
我們打開雨傘
索性塗黑了天空
--顧城