一种在NVcaffe框架下绘制曲线图的方法及装置制造方法及图纸

本发明专利技术公开了一种在NVcaffe框架下绘制曲线图的方法，通过调用plot函数库中的相关函数，实现将训练模型测试输出的日志中的目标变量的数据列表绘制于曲线图中，并通过plot函数库中的show函数显示绘制成的曲线图。相比于现有技术中调用NVcaffe自带的绘图工具，实现了在NVcaffe框架下显示训练、测试数据的曲线图，更方便用户对训练模型进行调整。本发明专利技术还提供一种在NVcaffe框架下绘制训练数据曲线图的装置及计算机可读存储介质，具有上述有益效果，在此不再赘述。

A method and device for drawing curves in NVcaffe framework

The invention discloses a method of drawing curves under the NVcaffe framework. By calling the related functions in the plot function library, the data list of the target variables in the log of the training model test output is plotted in the graph, and the plotted curves are displayed by the show function in the plot function library. Compared to the drawing tools that call NVcaffe in the existing technology, the curves of training and testing data in the NVcaffe framework are realized, and the user is more convenient to adjust the training model. The invention also provides a device for drawing the training data graph under the NVcaffe frame and a computer readable storage medium, which has the beneficial effect, and is no longer described here.

【技术实现步骤摘要】
一种在NVcaffe框架下绘制曲线图的方法及装置
本专利技术涉及计算机软件领域，特别是涉及一种在NVcaffe框架下绘制训练数据曲线图的方法、装置及计算机可读存储介质。
技术介绍
NVCaffe是NVIDIA基于BVLC-Caffe针对NVIDIAGPU，尤其是多GPU加速优化的开源深度学习框架。在使用NVCaffe进行训练数据、测试模型，当迭代次数非常大的时候，难免会想图形化展示实验结果，这样既便于训练过程中参数的调整，也便于最后成果的展示。在现有技术中，用户可以通过调用NVCaffe提供的绘图工具绘制训练数据的曲线图。但是NVCaffe没有提供显示曲线图的工具，用户只能在本地文件夹中查看曲线图，当用户需要根据曲线图调试模型参数时，需要在本地文件夹中找到曲线图的图片。因此，如何使用户能够直接在NVCaffe框架下查看训练、测试数据的曲线图，以方便调试参数，是本领域技术人员需要解决的技术问题。
技术实现思路
本专利技术的目的是提供一种在NVcaffe框架下绘制曲线图的方法、装置及计算机可读存储介质，用于使用户能够直接在NVCaffe框架下查看训练、测试数据的曲线图，以方便调试参数。为解决上述技术问题，本专利技术提供一种在NVcaffe框架下绘制曲线图的方法，包括：分析对训练模型测试输出的日志得到各目标变量的数据列表；调用plot函数库中的相关函数，依据所述数据列表将各所述目标变量绘制于曲线图上；调用所述plot函数库中的show函数显示所述曲线图。可选地，所述目标变量包括迭代次数或秒数中的任一项，以及Testaccuracy、Testloss、Trainlearningrate、Trainloss中的至少两项；所述调用plot函数库中的相关函数，依据所述数据列表将各所述目标变量绘制在曲线图上具体为：调用所述plot函数库中的相关函数，依据所述数据列表将各所述目标变量绘制于同一张所述曲线图中。可选地，所述分析所述日志得到各目标变量的数据列表具体为：根据各所述目标变量编写正则表达式，由所述正则表达式解析所述训练日志，以提取所述数据列表。可选地，所述plot函数库具体为Matplotlib.pyplot中的plot函数库。可选地，在显示所述曲线图之前，还包括：检查并删除所述曲线图上多余的点。可选地，在显示所述曲线图之后，还包括：保存所述曲线图。可选地，在分析所述日志之前，还包括：在Centos系统上部署NVCaffe的测试环境；设置相关参数，启动对训练模型的测试；保存测试输出的所述日志。可选地，所述相关参数包括取点间距、最大迭代次数和每批处理样本的个数。为解决上述技术问题，本专利技术还提供一种在NVcaffe框架下绘制训练数据曲线图的装置，包括：存储器，用于存储指令，所述指令包括如上述任一项所述在NVcaffe框架下绘制训练数据曲线图的方法的步骤；处理器，用于执行所述存储器中的指令。为解决上述技术问题，本专利技术还提供一种计算机可读存储介质，其上存储有计算机程序，所述计算机程序被处理器执行时实现如上述任一项所述在NVcaffe框架下绘制训练数据曲线图的方法的步骤。本专利技术提供的在NVcaffe框架下绘制曲线图的方法，通过调用plot函数库中的相关函数，实现将训练模型测试输出的日志中的目标变量的数据列表绘制于曲线图中，并通过plot函数库中的show函数显示绘制成的曲线图。相比于现有技术中调用NVcaffe自带的绘图工具，实现了在NVcaffe框架下显示训练、测试数据的曲线图，更方便用户对训练模型进行调整。本专利技术还提供一种在NVcaffe框架下绘制训练数据曲线图的装置及计算机可读存储介质，具有上述有益效果，在此不再赘述。附图说明为了更清楚的说明本专利技术实施例或现有技术的技术方案，下面将对实施例或现有技术描述中所需要使用的附图作简单的介绍，显而易见地，下面描述中的附图仅仅是本专利技术的一些实施例，对于本领域普通技术人员来讲，在不付出创造性劳动的前提下，还可以根据这些附图获得其他的附图。图1为本专利技术实施例提供的第一种在NVcaffe框架下绘制曲线图的方法的流程图；图2为本专利技术实施例提供的第二种在NVcaffe框架下绘制曲线图的方法的流程图；图3为本专利技术实施例提供的第三种在NVcaffe框架下绘制曲线图的方法的流程图；图4为本专利技术实施例提供的一种在NVcaffe框架下绘制训练数据曲线图的装置的结构示意图。具体实施方式本专利技术的核心是提供一种在NVcaffe框架下绘制曲线图的方法、装置及计算机可读存储介质，用于使用户能够直接在NVCaffe框架下查看训练、测试数据的曲线图，以方便调试参数。下面将结合本专利技术实施例中的附图，对本专利技术实施例中的技术方案进行清楚、完整地描述，显然，所描述的实施例仅仅是本专利技术一部分实施例，而不是全部的实施例。基于本专利技术中的实施例，本领域普通技术人员在没有做出创造性劳动前提下所获得的所有其他实施例，都属于本专利技术保护的范围。图1为本专利技术实施例提供的第一种在NVcaffe框架下绘制曲线图的方法的流程图。如图1所示，在NVcaffe框架下绘制曲线图的方法包括：S10：分析对训练模型测试输出的日志得到各目标变量的数据列表；S11：调用plot函数库中的相关函数，依据数据列表将各目标变量绘制于曲线图上；S12：调用plot函数库中的show函数显示曲线图。对于步骤S10来说，对训练模型测试输出的日志可以来自于采用预先保存在文件夹中的日志，也可以在NVcaffe框架下对训练模型进行测试后调用输出的日志。对训练模型进行训练具体可包括：在Centos系统上部署NVCaffe的测试环境；设置相关参数，启动对训练模型的测试；保存测试输出的日志。其中，相关参数可以包括取点间距、最大迭代次数和每批处理样本的个数(一个参考值包括取点间距为100，最大迭代次数为6000，每批处理样本的个数为128)。启动测试并保留测试输出的日志可通过如下设置实现：--solver＝Vgg-face/codes/solver.prototxt--gpu＝02>&1|tee/mnt/caffe/log/alexnet.test.log目标变量中的横坐标可以在迭代次数或秒数中选择，纵坐标可以在Testaccuracy、Testloss、Trainlearningrate、Trainloss中选择。可以根据所选的目标变量编写正则表达式，由正则表达式解析训练日志以提取数据列表，并将数据列表以数组的形式输出。对于步骤S11来说，可以采用matlab进行绘图，也可以采用python下的matplotlib进行绘图。通过预先试验，得到python下的matplotlib画出来的曲线图更为清晰，效果更好，因此可选用plot函数库具体为pathon的Matplotlib.pyplot中的plot函数库。对于步骤S12来说，调用plot函数库中的show函数，即在NVCaffe的指令输入窗口输入调用show函数的指令，即可显示曲线图，以便测试人员根据曲线图对训练模型进行改进。本专利技术实施例提供的在NVcaffe框架下绘制曲线图的方法，通过调用plot函数库中的相关函数，实现将训练模型测试输出的日志中的目标变量的数据列表绘制于曲线图中，并通过plot函数库中的show函本文档来自技高网...

【技术保护点】
1.一种在NVcaffe框架下绘制曲线图的方法，其特征在于，包括：分析对训练模型测试输出的日志得到各目标变量的数据列表；调用plot函数库中的相关函数，依据所述数据列表将各所述目标变量绘制于曲线图上；调用所述plot函数库中的show函数显示所述曲线图。

【技术特征摘要】
1.一种在NVcaffe框架下绘制曲线图的方法，其特征在于，包括：分析对训练模型测试输出的日志得到各目标变量的数据列表；调用plot函数库中的相关函数，依据所述数据列表将各所述目标变量绘制于曲线图上；调用所述plot函数库中的show函数显示所述曲线图。2.根据权利要求1所述的方法，其特征在于，所述目标变量包括迭代次数或秒数中的任一项，以及Testaccuracy、Testloss、Trainlearningrate、Trainloss中的至少两项；所述调用plot函数库中的相关函数，依据所述数据列表将各所述目标变量绘制在曲线图上具体为：调用所述plot函数库中的相关函数，依据所述数据列表将各所述目标变量绘制于同一张曲线图中。3.根据权利要求1所述的方法，其特征在于，所述分析所述日志得到各目标变量的数据列表具体为：根据各所述目标变量编写正则表达式，由所述正则表达式解析所述训练日志，以提取所述数据列表。4.根据权利要求1所述的方法，其特征在于，所述plot函数库具体为Matplotlib.p...

【专利技术属性】
技术研发人员：辛永欣，
申请(专利权)人：郑州云海信息技术有限公司，
类型：发明
国别省市：河南,41