Default threshold of 0.5

sepinf-inc · Feb 20, 2025 · eec717b · eec717b
1 parent d9020ca
commit eec717b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/iped-app/resources/config/conf/RemoteImageClassifier.txt b/iped-app/resources/config/conf/RemoteImageClassifier.txt
@@ -9,4 +9,4 @@ url = http://192.168.160.208:8000/zip
 batch_size = 50
 
 # threshold used to decide if an image is labeled in one category
-categorization_threshold=0.99
+categorization_threshold=0.50
diff --git a/iped-app/resources/scripts/tasks/AIRefineCategoryTask.js b/iped-app/resources/scripts/tasks/AIRefineCategoryTask.js
@@ -6,7 +6,6 @@
 /* Name of processing task
 */
 
-var categorizationThreshold=0.5;
 var RemoteImageClassifierConfig=Java.type("iped.engine.config.RemoteImageClassifierConfig")
 var Arrays=Java.type("java.util.Arrays")
 
@@ -21,7 +20,7 @@ function getConfigurables() {
 }
 
 function init(configurationManager) {
-	config=configurationManager.findObject(RemoteImageClassifierConfig.class)
+	var config=configurationManager.findObject(RemoteImageClassifierConfig.class)
 	categorizationThreshold=config.getCategorizationThreshold()	
 }