利用caffe建立自己的工程

     如果想把caffe应用到实际的工程当中,我们需要自己建立工程。这个真的是把我虐哭了,各种错误!最后还好全都解决了。在这个还有感谢happynea大神。

    至于如何建立工程,添加各种库我就不说了,我只说遇到的最要命的一个问题及解决办法!

    至于缺少库,自己百度解决吧,我说的是层的注册问题,第一次编译连接全都没错,运行时报错,unkown layer type  ,这是因为我们没有注册层,编译不知道层的存在。解决方法:https://github.com/happynear/caffe-windows/blob/master/FAQ.md


按照方面说的两种方法都能解决。

    但是如果你用CPU_ONLY版本时,出现的问题和这个相反,是层类型多次注册了。哎正反被虐。这种情况下第一种方法能够解决问题,但是第二种方法不能解决问题,这时就需要我们修改几行代码了。

找到layer_factory.hpp文件

// Adds a creator.
  static void AddCreator(const string& type, Creator creator) {
    CreatorRegistry& registry = Registry();
    //CHECK_EQ(registry.count(type), 0)
       // << "Layer type " << type << " already registered.";
	//下面是段测试程序
	if (registry.count(type) == 0)
	{
		registry[type] = creator;
	}
    //registry[type] = creator;
  }
找到solver_factory.hpp

// Adds a creator.
  static void AddCreator(const string& type, Creator creator) {
    CreatorRegistry& registry = Registry();
    //CHECK_EQ(registry.count(type), 0)
        //<< "Solver type " << type << " already registered.";
	if (registry.count(type) == 0)
	{
		registry[type] = creator;
	}
    //registry[type] = creator;
  }
修改代码后的稳定性,本人概不负责!如果你有好的方法,请留言回复,本人不胜感激!


评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值