封装caffe-windows-gpu(支持模型有多个输出)

之前封装的代码只能有一个输出：http://blog.csdn.net/sinat_30071459/article/details/51823390

并且有点问题，现在作了修改，模型可以有多个输出（当然也可以只有一个输出），可用于做图片的多属性分类，如：

这个例子有四个输出；

代码输出的分类结果为：

[plain] view plaincopy

typedef std::pair<int, float> Prediction;
vector<vector<Prediction> > predictions = classifier.Classify(img, 2);

int为标签下标（0，1，2，3...），float为确信度。predictions为模型所有输出的结果，每个结果取确信度排前二的情况。

如果想显示具体的标签，可自行修改代码。

代码下载地址：http://download.csdn.net/detail/sinat_30071459/9715053

封装caffe-windows-gpu(支持模型有多个输出)相关推荐

封装caffe-windows-cpu(支持模型有多个输出)
注意:该版本为CPU版本. 用到的caffe-windows来自:https://github.com/happynear/caffe-windows 先下载caffe-windows,解压:然后下载 ...
Linux拜拜！微软给WSL加入GPU支持，Windows终于迎来命令行包管理工具
点击上方"AI遇见机器学习",选择"星标"公众号重磅干货,第一时间送达白交发自凹非寺量子位报道 | 公众号 QbitAI 看完昨晚微软Build大 ...
【caffe-Windows】caffe+VS2013+Windows+GPU配置+cifar使用
前言国际惯例,先来波地址: CUDA WIN7:链接:http://pan.baidu.com/s/1nvyA3Qp 密码:h0f3 官方网址:https://developer.nvidia. ...
【理论恒叨】【立体匹配系列】经典PatchMatch: （1）Slanted support windows倾斜支持窗模型
一枝独秀不是春理论恒叨系列 [理论恒叨][立体匹配系列]经典PatchMatch: (1)Slanted support windows倾斜支持窗模型 [理论恒叨][立体匹配系列]经典PatchMa ...
Caffe Windows版本的编译
2019独角兽企业重金招聘Python工程师标准>>> 1:Caffe的主版本只支持Linux,所以要下载专门的Caffe Windows版本,网址为 https://github. ...
Pytorch基础训练库Pytorch-Base-Trainer(支持模型剪枝分布式训练)
Pytorch基础训练库Pytorch-Base-Trainer(支持模型剪枝分布式训练) 目录 Pytorch基础训练库Pytorch-Base-Trainer(PBT)(支持分布式训练) 1.I ...
微软开源 TensorFlow-DirectML，为 WSL2 提供 GPU 支持
微软近日开源了 TensorFlow-DirectML 项目,它是 Tensorflow 的一个分支,利用 DirectML 在 Windows 10 和 WSL 上提供硬件加速. DirectML ...
Caffe学习笔记2--Ubuntu 14.04 64bit 安装Caffe（GPU版本）
0.检查配置 1. VMWare上运行的Ubuntu,并不能支持真实的GPU(除了特定版本的VMWare和特定的GPU,要求条件严格,所以我在VMWare上搭建好了Caffe环境后,又重新在Windo ...
3月13日云栖精选夜读：通过阿里云容器服务深度学习解决方案上手Caffe+多GPU训练
阿里云容器服务提供的深度学习解决方案内置了对Tensorflow, Keras, MXnet框架的环境,并支持基于它们的深度学习模型开发.模型训练和模型预测.同时,对于模型训练和预测,用户还可以通过指 ...

封装caffe-windows-gpu(支持模型有多个输出)

封装caffe-windows-gpu(支持模型有多个输出)相关推荐

最新文章

热门文章