BVLC#38: pre-class accuracy fix

anandthakker · Aug 2, 2016 · cd99f1e · cd99f1e
1 parent 54e0c9f
commit cd99f1e
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/src/caffe/layers/accuracy_layer.cpp b/src/caffe/layers/accuracy_layer.cpp
@@ -95,7 +95,17 @@ void AccuracyLayer<Dtype>::Forward_cpu(const vector<Blob<Dtype>*>& bottom,
   top[0]->mutable_cpu_data()[0] = accuracy / count;
   if (top.size() >= 2) {
     for (int i = 0; i < num_labels; ++i) {
-      top[1]->mutable_cpu_data()[i] = accuracies[i] / nums[i];
+      if (nums[i] == 0) {
+      	if (accuracies[i] == 0) {
+          top[1]->mutable_cpu_data()[i] = 1;
+        }
+        else {
+          top[1]->mutable_cpu_data()[i] = 0;
+        }
+      }
+      else {
+      	top[1]->mutable_cpu_data()[i] = accuracies[i] / nums[i];
+      }
     }
   }
   // Accuracy layer should not be used as a loss function.