PS:本文主要用于自我整理总结,方便后续参考,如果恰好帮助到你,也是件值得高兴的事
我们运行生成的模型命名默认应该是这样的:
它每1000步保存一次,这有点久,如果中间有问题,比如出现nan,那前面的几百步都白跑了,所以想到了要修改它保存的步数,同时,我还想保存的时候顺便显示其loss时,方便我进行选择。
步骤如下:
首先我们使用VS打开darknet.sln文件。
然后选择detector.c文件,进行编辑。
我的是在390行上下进行修改。①386行的是修改保存步数,我将之前的1000改成100;②392行是在保存时加了一个显示loss值的操作
最后,修改完后,重新生成解决方案,重新生成darknet.exe即可。
这样我们生成的权重模型则就可以变成这样:
对darknet源码进行修改,改变保存模型的步数及命名
最新推荐文章于 2022-03-28 22:15:14 发布