[發明專利]一種在NVcaffe框架下繪制曲線圖的方法及裝置

在審

• 申請號: 201810161509.2
• 申請日: 2018-02-27
• 公開（公告）號: CN108268362A
• 公開（公告）日: 2018-07-10
• 發明（設計）人: 辛永欣
• 申請（專利權）人: 鄭州云海信息技術有限公司
• 主分類號: G06F11/32
• 分類號: G06F11/32;G06T11/20
• 代理公司: 北京集佳知識產權代理有限公司 11227
• 代理人: 羅滿

本發明公開了一種在NVcaffe框架下繪制曲線圖的方法，通過調用plot函數庫中的相關函數，實現將訓練模型測試輸出的日志中的目標變量的數據列表繪制于曲線圖中，并通過plot函數庫中的show函數顯示繪制成的曲線圖。相比于現有技術中調用NVcaffe自帶的繪圖工具，實現了在NVcaffe框架下顯示訓練、測試數據的曲線圖，更方便用戶對訓練模型進行調整。本發明還提供一種在NVcaffe框架下繪制訓練數據曲線圖的裝置及計算機可讀存儲介質，具有上述有益效果，在此不再贅述。

技術領域

本發明涉及計算機軟件領域，特別是涉及一種在NVcaffe框架下繪制訓練數據曲線圖的方法、裝置及計算機可讀存儲介質。

背景技術

NVCaffe是NVIDIA基于BVLC-Caffe針對NVIDIAGPU，尤其是多GPU加速優化的開源深度學習框架。在使用NVCaffe進行訓練數據、測試模型，當迭代次數非常大的時候，難免會想圖形化展示實驗結果，這樣既便于訓練過程中參數的調整，也便于最后成果的展示。

在現有技術中，用戶可以通過調用NVCaffe提供的繪圖工具繪制訓練數據的曲線圖。但是NVCaffe沒有提供顯示曲線圖的工具，用戶只能在本地文件夾中查看曲線圖，當用戶需要根據曲線圖調試模型參數時，需要在本地文件夾中找到曲線圖的圖片。

因此，如何使用戶能夠直接在NVCaffe框架下查看訓練、測試數據的曲線圖，以方便調試參數，是本領域技術人員需要解決的技術問題。

發明內容

本發明的目的是提供一種在NVcaffe框架下繪制曲線圖的方法、裝置及計算機可讀存儲介質，用于使用戶能夠直接在NVCaffe框架下查看訓練、測試數據的曲線圖，以方便調試參數。

為解決上述技術問題，本發明提供一種在NVcaffe框架下繪制曲線圖的方法，包括：

分析對訓練模型測試輸出的日志得到各目標變量的數據列表；

調用plot函數庫中的相關函數，依據所述數據列表將各所述目標變量繪制于曲線圖上；

調用所述plot函數庫中的show函數顯示所述曲線圖。

可選地，所述目標變量包括迭代次數或秒數中的任一項，以及Test accuracy、Test loss、Train learning rate、Train loss中的至少兩項；

所述調用plot函數庫中的相關函數，依據所述數據列表將各所述目標變量繪制在曲線圖上具體為：

調用所述plot函數庫中的相關函數，依據所述數據列表將各所述目標變量繪制于同一張所述曲線圖中。

可選地，所述分析所述日志得到各目標變量的數據列表具體為：

根據各所述目標變量編寫正則表達式，由所述正則表達式解析所述訓練日志，以提取所述數據列表。

可選地，所述plot函數庫具體為Matplotlib.pyplot中的plot函數庫。

可選地，在顯示所述曲線圖之前，還包括：

檢查并刪除所述曲線圖上多余的點。

可選地，在顯示所述曲線圖之后，還包括：

保存所述曲線圖。

可選地，在分析所述日志之前，還包括：

在Centos系統上部署NVCaffe的測試環境；

設置相關參數，啟動對訓練模型的測試；

保存測試輸出的所述日志。

可選地，所述相關參數包括取點間距、最大迭代次數和每批處理樣本的個數。