Fix the zero output case handling

PiperOrigin-RevId: 276485691
tensorflow · Oct 24, 2019 · 4fcc9a4 · 4fcc9a4
1 parent 348ea2d
commit 4fcc9a4
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/lite/examples/object_detection/ios/ObjectDetection/ModelDataHandler/ModelDataHandler.swift b/lite/examples/object_detection/ios/ObjectDetection/ModelDataHandler/ModelDataHandler.swift
@@ -190,6 +190,9 @@ class ModelDataHandler: NSObject {
   /// sorted in descending order.
   func formatResults(boundingBox: [Float], outputClasses: [Float], outputScores: [Float], outputCount: Int, width: CGFloat, height: CGFloat) -> [Inference]{
     var resultsArray: [Inference] = []
+    if (outputCount == 0) {
+      return resultsArray
+    }
     for i in 0...outputCount - 1 {
 
       let score = outputScores[i]