How to get the confidence of my prediction? about caffe-android-lib

[carlosporta]

I added to caffe_jni.cpp the following lines:
JNIEXPORT jfloatArray JNICALL
Java_com_sh1r0_caffe_1android_1lib_CaffeMobile_getConfidence(JNIEnv *env,
jobject thiz,
jstring imgPath) {
CaffeMobile *caffe_mobile = CaffeMobile::Get();
vector confidence = caffe_mobile->GetConfidence(jstring2string(env, imgPath));

jfloatArray result;
result = env->NewFloatArray(4);
if (result == NULL) {
return NULL; /* out of memory error thrown */
}
// move from the temp structure to the java structure
env->SetFloatArrayRegion(result, 0, 4, &confidence[0]);
return result;
}

Then added to caffe_mobile.cpp

vector CaffeMobile::GetConfidence(const string &img_path) {
const vector probs = Forward(img_path);
return probs;
}

And created a native method on android. But it returns the following code ( I have 4 classes):
label 0 = 0.0
label 1 = 1.0
label 2 = 0.0
label 3 = 0.0

Another prediction:
label 0 = 0.0
label 0 = 0.0
label 0 = 0.0
label 0 = 1.0

So, it gives not gives to me the confidence that I want. Am I doing something wrong? Or is not possible?

[g1910]

Hi,
We are trying to build an app which requires to classify objects based on whether the confidence value is higher than a threshold. This work of yours looks really promising and we would like to go forward with using it. Problem is that we really need the individual confidence values. Please see if this functionality can be added asap.

Else, it seems that only the .so is accessible in the recent commits and not the .cpp code which we can ourselves modify. See if you can re-add in the upcoming commits.

[sh1r0]

I have updated the dev branch with GetConfidenceScore function in caffe_mobile module. The source files are right here.
Please have a try and kindly let me know if there is any problem. Thanks.

[g1910]

I can see the changes in the .cpp files. But, how to get them reflected in the java class?

Please update the caffe-android-demo project with these changes. Also, please add some documentation on how to get the functionality from .cpp to .java.