add list of occ in meta

utils.py

@@ -40,20 +40,33 @@ def compute_id_good_occ(data_name, count, categories_instance_array_id_to_cat, c
     f2 = open(f'{data_name}/Occlusion/{count}.json')
 
     data2 = json.load(f2)
-    categories_occ_array = {}
-    categories_array = {}
+    categories_array_filtered = {}
+    categories_array_filtered_occ = {}
+
+    categories_array_all = {}
+    categories_array_all_occ = {}
 
     for cat in categories_instance_array_cat_to_id :
-        categories_occ_array[cat] = []
-        categories_array[cat] = []
+        categories_array_filtered[cat] = []
+        categories_array_filtered_occ[cat] = []
+        categories_array_all[cat] = []
+        categories_array_all_occ[cat] = []
 
     for i in data2:
         if i['occlusion_value'] >= Occ_wanted :
-            categories_occ_array[categories_instance_array_id_to_cat[i['id']]].append(i['id'])
-        elif  i['occlusion_value'] >= 0.05  and i['occlusion_value'] < Occ_wanted:
-            categories_array[categories_instance_array_id_to_cat[i['id']]].append(i['id'])
+            categories_array_filtered[categories_instance_array_id_to_cat[i['id']]].append(i['id'])
+            categories_array_filtered_occ[categories_instance_array_id_to_cat[i['id']]].append(i['occlusion_value'])
+        if  i['occlusion_value'] >= 0.05 :
+            categories_array_all[categories_instance_array_id_to_cat[i['id']]].append(i['id'])
+            categories_array_all_occ[categories_instance_array_id_to_cat[i['id']]].append(i['occlusion_value'])
 
     # Closing file
     f2.close()
 
-    return categories_occ_array, categories_array
+    #print(categories_array_filtered)
\ No newline at end of file
+    #print(categories_array_filtered_occ)
+    #print(categories_array_all)
+    #print(categories_array_all_occ)
+
+
+    return categories_array_filtered, categories_array_filtered_occ, categories_array_all, categories_array_all_occ
\ No newline at end of file