From 87cb0ae9b9a8a0628308a0b865f83f42ae7c5d2d Mon Sep 17 00:00:00 2001
From: mrt <marcos.rodrigo@upm.es>
Date: Tue, 12 Sep 2023 14:53:43 +0200
Subject: [PATCH] Remove any person which does not have images for all
 categories

---
 real_world_occluded_faces_test.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/real_world_occluded_faces_test.py b/real_world_occluded_faces_test.py
index a9e869f..909cdfb 100644
--- a/real_world_occluded_faces_test.py
+++ b/real_world_occluded_faces_test.py
@@ -47,6 +47,14 @@ def get_histogram():
     for item in sunglasses_items:
         name = item.split('_wearing_sunglasses')[0]
         data[name]['Sunglasses'] = len(os.listdir(f"{SUNGLASSES_DIR}/{item}"))
+
+    # Remove any person which does not have images for all categories
+    data_aux = data.copy()
+    for name in data.keys():
+        if any(val is None for val in data[name].values()):
+            data_aux.pop(name)
+    data = data_aux.copy()
+
     return data