深度学习
文章平均质量分 55
wudi_X
学习使人快乐!
展开
-
【caffe】SSD中的default box的可视化
前言博主在测试分析目标检测网络SSD的时候,想知道每个检测框是来自哪个default box的,因此需要对SSD源码做适当修改,使其在测试时输出相应的信息。SSD算法的介绍有很多博客讲得很好,这里不多说了。这篇博客主要记录了如何用caffe的python接口测试SSD并获取SSD预测框的default box的索引。具体步骤1. 源码修改首先修改detection_outp...原创 2018-03-23 22:18:08 · 2417 阅读 · 6 评论 -
caffe缺少NCCL库导致不能多GPU训练问题(改makefile版)
最近用caffe多GPU联调时遇到问题,如果正常用caffe的多gpu训练而不加初始化模型,末尾加--gpu=0,1,不会有问题;然而当用pretrained模型finetune网络时,如--weights=xxx.caffemodel,则会报如下错误: Multi-GPU execution not available - rebuild with USE_NCCL 提示nccl没有安装,于...原创 2018-04-19 23:31:37 · 3036 阅读 · 2 评论 -
【caffe】修改SSD的solver输出recall和precision
我们可以从SSD的caffe源码中得到test的mAP,mAP是不同阈值下的precision均值,但如果我们想看某个阈值下的recall和precision时,就需要对solver.cpp源码做一定修改。 关于mAP, recall和precision的解释这里不赘述,可以参考以下博客: 中文介绍 英文介绍修改caffe.proto首先在src/caffe/proto/caffe....原创 2018-05-23 11:08:13 · 3109 阅读 · 3 评论