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

本文介绍了一个改进后的图片分类模型,该模型能够同时输出多个分类结果,并提供了代码实现及下载链接。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

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

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


这个例子有四个输出;

代码输出的分类结果为:

[plain]  view plain  copy
  1. typedef std::pair<int, float> Prediction;  
  2. vector<vector<Prediction> > predictions = classifier.Classify(img, 2);  
int为标签下标(0,1,2,3...),float为确信度。predictions为模型所有输出的结果,每个结果取确信度排前二的情况。

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

代码下载地址:http://download.youkuaiyun.com/detail/sinat_30071459/9715053

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值