Use the same weighted score mean of the DIE Task

sepinf-inc · Feb 25, 2025 · 4072a14 · 4072a14
1 parent c1673aa
commit 4072a14
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/iped-engine/src/main/java/iped/engine/task/RemoteImageClassifier.java b/iped-engine/src/main/java/iped/engine/task/RemoteImageClassifier.java
@@ -61,12 +61,11 @@ public void addClass(String classname,double val) {
         }
 
         public double getClassProb(String classname) {
-            double resp = 0;
-            for (Double val : classes.get(classname)) {
-                if (val > resp)
-                    resp = val;
-            }
-            return resp;
+            /*
+             * double resp = 0; for (Double val : classes.get(classname)) { if (val > resp)
+             * resp = val; } return resp;
+             */
+            return DIETask.videoScore(classes.get(classname));
 
         }
     }