Added auto softmax metadata for action_recognition (#969)

deepjavalibrary · May 20, 2021 · f24dad9 · f24dad9
1 parent d7ef4ec
commit f24dad9
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/...esources/mlrepo/model/cv/action_recognition/ai/djl/mxnet/action_recognition/metadata.json b/...esources/mlrepo/model/cv/action_recognition/ai/djl/mxnet/action_recognition/metadata.json
@@ -27,7 +27,8 @@
         "height": 224,
         "resize": true,
         "normalize": true,
-        "synsetFileName": "classes.txt"
+        "synsetFileName": "classes.txt",
+        "applySoftmax": true
       },
       "files": {
         "classes": {
@@ -60,7 +61,8 @@
         "height": 299,
         "resize": true,
         "normalize": true,
-        "synsetFileName": "classes.txt"
+        "synsetFileName": "classes.txt",
+        "applySoftmax": true
       },
       "files": {
         "classes": {