edge case for score none if no predictions

weecology · Jun 10, 2021 · fe29629 · fe29629
1 parent d8fc0d5
commit fe29629
Show file tree

Hide file tree

Showing 2 changed files with 86 additions and 89 deletions.
diff --git a/deepforest/evaluate.py b/deepforest/evaluate.py
@@ -86,7 +86,7 @@ def evaluate(predictions,
 
         #If empty, add to list without computing IoU
         if image_predictions.empty: 
-            result = pd.DataFrame({"truth_id":group.index.values,"prediction_id": None, "IoU":0, "predicted_label": None, "true_label":group.label})
+            result = pd.DataFrame({"truth_id":group.index.values,"prediction_id": None, "IoU":0, "predicted_label": None, "score":None, "true_label":group.label})
             results.append(result)
             continue
         else: