fix(tvm_utility): fix warning of negativeContainerIndex (#6924)

Signed-off-by: Ryuta Kambe <[email protected]>
autowarefoundation · May 8, 2024 · 6d00269 · 6d00269
1 parent b6d7477
commit 6d00269
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/common/tvm_utility/example/yolo_v2_tiny/main.cpp b/common/tvm_utility/example/yolo_v2_tiny/main.cpp
@@ -200,6 +200,8 @@ class PostProcessorYoloV2Tiny : public tvm_utility::pipeline::PostProcessor<std:
             }
           }
 
+          if (max_ind == -1) continue;
+
           // Decode and copy class probabilities
           std::vector<float> class_probabilities{};
           float p_total = 0;