Added check for existance of groundtruth_dict[standard_fields.InputDa…

…taFields.groundtruth_difficult]
gedefet · Nov 14, 2018 · 61b5a0c · 61b5a0c
1 parent 8a8d3bc
commit 61b5a0c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/research/object_detection/utils/object_detection_evaluation.py b/research/object_detection/utils/object_detection_evaluation.py
@@ -210,6 +210,7 @@ def add_single_ground_truth_image_info(self, image_id, groundtruth_dict):
     # use values from the dictionary or insert None otherwise.
     if (standard_fields.InputDataFields.groundtruth_difficult in
         groundtruth_dict.keys() and
+        groundtruth_dict[standard_fields.InputDataFields.groundtruth_difficult] and
         (groundtruth_dict[standard_fields.InputDataFields.groundtruth_difficult]
          .size or not groundtruth_classes.size)):
       groundtruth_difficult = groundtruth_dict[
@@ -473,6 +474,7 @@ def add_single_ground_truth_image_info(self, image_id, groundtruth_dict):
     # use values from the dictionary or insert None otherwise.
     if (standard_fields.InputDataFields.groundtruth_group_of in
         groundtruth_dict.keys() and
+        groundtruth_dict[standard_fields.InputDataFields.groundtruth_group_of] and
         (groundtruth_dict[standard_fields.InputDataFields.groundtruth_group_of]
          .size or not groundtruth_classes.size)):
       groundtruth_group_of = groundtruth_dict[