标签: computer-vision classification caffe pycaffe
我正在尝试使用Caffe构建二元分类器。首先,我通过稍微修改微调样式识别模型来训练我自己的图像数据集。但是当我尝试对单个图像进行分类时,它给出了两个类的概率为50%。为了分类,我使用了相同的deploy.prototxt,只是将output_number更改为2.这很奇怪,因为在训练时我得到准确率~85%。我尝试了python(classify.py)和cpp(classification.cpp),两者都给出了相同的结果。我想我在管道上做错了。
感谢您的帮助。