Log warn when confidence is not equal to size of objects.
Project: http://git-wip-us.apache.org/repos/asf/tika/repo Commit: http://git-wip-us.apache.org/repos/asf/tika/commit/0096dd78 Tree: http://git-wip-us.apache.org/repos/asf/tika/tree/0096dd78 Diff: http://git-wip-us.apache.org/repos/asf/tika/diff/0096dd78 Branch: refs/heads/TIKA-1508 Commit: 0096dd78c93840091537ed9324d24d2e9c5b9f28 Parents: a1d1a81 Author: Chris Mattmann <[email protected]> Authored: Sun Aug 14 09:00:29 2016 -0700 Committer: Chris Mattmann <[email protected]> Committed: Sun Aug 14 09:00:29 2016 -0700 ---------------------------------------------------------------------- .../tika/parser/recognition/tf/TensorflowRESTRecogniser.java | 3 +++ 1 file changed, 3 insertions(+) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/tika/blob/0096dd78/tika-parsers/src/main/java/org/apache/tika/parser/recognition/tf/TensorflowRESTRecogniser.java ---------------------------------------------------------------------- diff --git a/tika-parsers/src/main/java/org/apache/tika/parser/recognition/tf/TensorflowRESTRecogniser.java b/tika-parsers/src/main/java/org/apache/tika/parser/recognition/tf/TensorflowRESTRecogniser.java index 8722dee..8085df6 100644 --- a/tika-parsers/src/main/java/org/apache/tika/parser/recognition/tf/TensorflowRESTRecogniser.java +++ b/tika-parsers/src/main/java/org/apache/tika/parser/recognition/tf/TensorflowRESTRecogniser.java @@ -119,6 +119,9 @@ public class TensorflowRESTRecogniser implements ObjectRecogniser { JSONObject jReply = new JSONObject(replyMessage); JSONArray jClasses = jReply.getJSONArray("classnames"); JSONArray jConfidence = jReply.getJSONArray("confidence"); + if (jClasses.length() != jConfidence.length()){ + LOG.warn("Classes of size {} is not equal to confidence of size {}", jClasses.length(), jConfidence.length()); + } assert jClasses.length() == jConfidence.length(); for (int i = 0; i < jClasses.length(); i++) { RecognisedObject recObj = new RecognisedObject(jClasses.getString(i),
