fix: dont compute highest if nothing detected

HiroIshida · Dec 31, 2023 · 13eacf7 · 13eacf7
1 parent 6121b98
commit 13eacf7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/node_script/wrapper.py b/node_script/wrapper.py
@@ -123,7 +123,7 @@ def infer(self, msg: Image) -> InferenceRawResult:
         scores = instances.scores.tolist()
         class_indices = instances.pred_classes.tolist()
 
-        if self.node_config.output_highest:
+        if len(scores) > 0 and self.node_config.output_highest:
             best_index = np.argmax(scores)
             pred_masks = [pred_masks[best_index]]
             scores = [scores[best_index]]