在跑DenseNet代码的时候遇到的问题

去年新出来一个DenseNet,据说效果很好,在下载好caffemodel+deploy.prototxt之后,运行predict.py出现了如下的错误

Question:https://github.com/shicai/DenseNet-Caffe/issues/10

WARNING: Logging before InitGoogleLogging() is written to STDERR
W0805 10:26:38.720525 3101 _caffe.cpp:139] DEPRECATION WARNING - deprecated use of Python interface
W0805 10:26:38.720546 3101 _caffe.cpp:140] Use this instead (with the named "weights" parameter):
W0805 10:26:38.720549 3101 _caffe.cpp:142] Net('/home/xuy/桌面/code/python/caffe/Bag_gender_hair_classification/DenseNet_deploy_161.prototxt', 1, weights='/home/xuy/桌面/code/python/caffe/Bag_gender_hair_classification/model/DenseNet_161.caffemodel')
[libprotobuf ERROR google/protobuf/text_format.cc:245] Error parsing text-format caffe.NetParameter: 54:14: Message type "caffe.PoolingParameter" has no field named "ceil_mode".
F0805 10:26:38.740175 3101 upgrade_proto.cpp:88] Check failed: ReadProtoFromTextFile(param_file, param) Failed to parse NetParameter file: /home/xuy/桌面/code/python/caffe/Bag_gender_hair_classification/DenseNet_deploy_161.prototxt
*** Check failure stack trace: ***


主要的问题是:ceil_mode的问题,因此需要对于caffe源代码进行修改


解决办法:

https://github.com/BVLC/caffe/pull/3057/files


其中,include/caffe/vision_layers.hpp,由于caffe结构已经修改,没有这个文件了,经过查看之后,发现该文件是:/home/xuy/caffe/include/caffe/layers/pooling_layer.hpp


在这里面进行修改即可


其他的照着代码加即可





  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 5
    评论
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值