yolov3修改保存weights间隔

最新推荐文章于 2023-01-12 15:35:47 发布

3DYour

最新推荐文章于 2023-01-12 15:35:47 发布

阅读量936

点赞数 2

分类专栏：笔记文章标签： python

本文链接：https://blog.csdn.net/abc1831939662/article/details/118785517

版权

笔记专栏收录该内容

18 篇文章 1 订阅

订阅专栏

这篇博客介绍了如何修改Yolov3的darknet源码，以改变权重备份的间隔。通过在detector.c文件中查找并修改保存权重的条件语句，将原本每10000次或训练前3000次每100次保存一次权重的设置，改为每1000次保存一次。修改后重新编译并运行darknet，以实现新的权重保存策略。

摘要由CSDN通过智能技术生成

yolov3修改保存weights间隔

打开darknet/src/detector.c
搜索‘save’
找到

 if(i%100==0){
#ifdef GPU
            if(ngpus != 1) sync_nets(nets, ngpus, 0);
#endif
            char buff[256];
            sprintf(buff, "%s/%s.backup", backup_directory, base);
            save_weights(net, buff);
        }
        if(i%1000 == 0){// if(i%10000==0 || (i < 3000 && i%100 == 0))
#ifdef GPU
            if(ngpus != 1) sync_nets(nets, ngpus, 0);
#endif
            char buff[256];
            sprintf(buff, "%s/%s_%d.weights", backup_directory, base, i);
            save_weights(net, buff);
        }
        free_data(train);
    }

把上面的