From 4072a149ed957d1243544439dbbac77fc5bbef79 Mon Sep 17 00:00:00 2001 From: Hauck Date: Tue, 25 Feb 2025 13:38:28 -0300 Subject: [PATCH] Use the same weighted score mean of the DIE Task --- .../java/iped/engine/task/RemoteImageClassifier.java | 11 +++++------ 1 file changed, 5 insertions(+), 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 index f2cbcd73fe..95be6113e9 100644 --- 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)); } }