使用classification.cpp对图像进行分类时Caffe出错

时间:2016-04-26 08:29:00

标签: c++ caffe conv-neural-network

我已经训练了自己的网络。培训很好。我也使用了'caffe time',它正常估计前进和后退的时间。但是当我运行时:(Using this ref)

./build/examples/cpp_classification/classification.bin models/own_net/deploy.prototxt examples/RSR_50k_all_1k_db/snapshot_iter_10000.caffemodel examples/RSR_50k_all_1k_db/mean.binaryproto examples/RSR_50k_all_1k_db/labels.txt /home/ubuntu/datasets/RSR_50k_1ll_1k/Testing/[0]/outfile243.jpg

这会产生错误:

F0426 10:10:50.063822  2714 classification.cpp:63] Check failed: net_->num_outputs() == 1 (2 vs. 1) Network should have exactly one output.
*** Check failure stack trace: ***
    @ 0xf6c5d060  (unknown)
    @ 0xf6c5cf5c  (unknown)
    @ 0xf6c5cb78  (unknown)
    @ 0xf6c5ef98  (unknown)
    @     0xd10c  Classifier::Classifier()
    @     0xb0a2  main
    @ 0xf672c632  (unknown)
Aborted

当我使用相同的命令使用caffenet对股票猫图像进行分类时,它可以正常工作。我怀疑标签文件有问题。我的标签文件仅列出所有标签,每行一个。知道我做错了吗?

不幸的是,网不是我的,所以我不认为我可以分享完整的结构。但是,它有一些conv,relu和fc图层,并以此图层结束:

layer {
  name: "prob"
  type: "Softmax"
  bottom: "ip3"
  top: "prob"
}

我怀疑可能是罪魁祸首。

0 个答案:

没有答案