瀏覽代碼

kleine Aenderung bei der Auswertung

Bjoern Froehlich 13 年之前
父節點
當前提交
d8728e7f58
共有 1 個文件被更改,包括 9 次插入1 次删除
  1. 9 1
      progs/testSemanticSegmentation.cpp

+ 9 - 1
progs/testSemanticSegmentation.cpp

@@ -269,7 +269,15 @@ int main( int argc, char **argv )
     if (( classNames.existsClassno( r ) ) && ( forbidden_classes.find( r ) == forbidden_classes.end() ) )
     {
       avg_perf += M( r, r );
-      classes_trained++;
+      double lsum = 0.0;
+      for(int r2 = 0; r2 < ( int )M.rows(); r2++)
+      {
+        lsum += M(r2,r);
+      }
+      if(lsum != 0.0)
+      {
+        classes_trained++;
+      }
     }
   }