永发信息网

在VS2013中打开caffe源代码,都能编译成功,下一步应该怎么训练模型

答案:1  悬赏:80  手机版
解决时间 2021-03-18 19:32
在VS2013中打开caffe源代码,都能编译成功,下一步应该怎么训练模型
最佳答案
你想调用你的模型,最简单的办法是看examples/cpp_classification里面的cpp文件,那是教你如何调用caffe获取分类结果的...(你没接触过caffe的话,建议你直接按照这个文件来操作可能会比较简单,下面我的代码我也不知道没接触过caffe的人看起来难度会有多大)

不过那个代码我看着不太习惯,所以之前自己稍微写了一个简易的版本,不知道怎么上传附件,懒人一个就直接把代码贴在最后了。
先简单解释一下如何使用,把这个代码复制到一个头文件中,然后放在examples里面一个自己创建的文件夹里面,然后写一个main函数调用这个类就可以了,比如:
复制,保存到caffe/examples/myproject/net_operator.hpp,然后同目录下写一个main.cpp,在main函数里面#include "net_operator.hpp",就可以使用这个类了:
const string net_prototxt = "..."; // 你的网络的prototxt文件,用绝对路径,下面同理
const string pre_trained_file = "..."; // 你训练好的.caffemodel文件
const string img_path = "..."; // 你要测试的图片路径
// 创建NetOperator对象
NetOperator net_operator(net_prototxt, pre_trained_file);
Blob *blob = net_operator.processImage(img_path);
// blob就得到了最后一层的输出结果,至于blob里面是怎么存放数据的,你需要去看看官网对它的定义

写完main.cpp之后,到caffe目录下,make,然后它会编译你写的文件,对应生成的可执行文件。比如按我上面写的那样,make之后就会在caffe/build/examples/myproject文件夹里面生成一个main.bin,执行这个文件就可以了。因为生成的可执行文件并不是直接在代码目录下,所以前面我建议你写的路径用绝对路径

另外如果你要获取的不是最后一层的输出,你需要修改一下processImage函数的返回值,通过NetOperator的成员变量net_来获取你需要的blob,比如有个blob名称为"label",你想获取这个blob,可以通过net_->blob_by_name("label")来获取,当然获取到的是shared_ptr >类型的,搜一下boost shared_ptr就知道跟普通指针有什么不同了
我要举报
如以上问答信息为低俗、色情、不良、暴力、侵权、涉及违法等信息,可以点下面链接进行举报!
大家都在看
天天吃五花肉会不会变胖啊?
催乳师怎么找工作
我前男友我跟他是因为父母反对才分手的,现在
我在昆明工地上做活,带着十几个人做了几个月
一农场邮政支局地址有知道的么?有点事想过去
空姐李倩现任老公是谁
请问有知道ViVOX9L手机屏幕坏了,内外屏一起
聪明小空空有哪些人物
盐城市玻璃钢制品厂地址有知道的么?有点事想
篆字拍照识别!多谢!
鸵鸟幼崽多少钱一只啊
自考高升专一般要多久呢?
office2013英文版为什么会显示中文版
我看好多加湿器都推出了各种防二次污染的方法
哪有卖牛肉沫的,喂狗的,狗挑食,不知道怎么
推荐资讯
阜阳师范学院专升本计算机专业的C语言课本是
临武县舜峰镇西城村卫生室在什么地方啊,我要
牛群和冯巩的那个关于男女的相声叫什么啊
圣安地列斯可以用单车拱墙然后在墙上跳吗,想
当人突然进入寒冷环境中时,相应的反应是( 
我明明已经获取root 可为什么用不了需root 的
想参加模拟联合国,需要准备哪些东西?有没有
做ghost一键备份时出现 Try(hd0,0):FAT32
三生三世里我最讨厌白浅了一点也不宽宏大量。
驾驶证只有三分了要扣十二分怎么办
红锦鲤(观赏鱼)如何能变得更红
大核桃和小核桃哪个营养价值高
正方形一边上任一点到这个正方形两条对角线的
阴历怎么看 ?