caffe
文章平均质量分 67
wudi_X
学习使人快乐!
展开
-
Caffe用python接口分类的例子
如何用训练好的caffemodel测试图片分类呢?caffe官方提供python的分类例子在examples/00-classification.ipynb下,但是只介绍了caffe.Net分类,这篇博客记录了另一中方式caffe.Classifier,并介绍了两者的区别。1. caffe.Net步骤大概分为:模型导入->数据处理->分类和输出模型导入需要deploy(网络定义)和ca...原创 2018-03-16 11:25:11 · 1046 阅读 · 0 评论 -
caffe工厂类solver_factory解析
caffe工厂类solver_factory根据solver.prototxt中定义的type类型创建各种solver实例创建代码如下:caffe.cpp shared_ptr<caffe::Solver<float> > solver(caffe::SolverRegistry<float>::CreateSolver(solver_param)...原创 2017-10-11 10:03:58 · 593 阅读 · 0 评论 -
【caffe】SSD中的default box的可视化
前言博主在测试分析目标检测网络SSD的时候,想知道每个检测框是来自哪个default box的,因此需要对SSD源码做适当修改,使其在测试时输出相应的信息。SSD算法的介绍有很多博客讲得很好,这里不多说了。这篇博客主要记录了如何用caffe的python接口测试SSD并获取SSD预测框的default box的索引。具体步骤1. 源码修改首先修改detection_outp...原创 2018-03-23 22:18:08 · 2384 阅读 · 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 · 2988 阅读 · 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 · 3049 阅读 · 3 评论