From 0f25276a992a829565f30027763fe194a0c465ed Mon Sep 17 00:00:00 2001
From: Gduret <guillaume.duret@ec-lyon.fr>
Date: Tue, 16 May 2023 22:51:06 +0200
Subject: [PATCH] add list of occ in meta
---
utils.py | 29 +++++++++++++++++++++--------
1 file changed, 21 insertions(+), 8 deletions(-)
diff --git a/utils.py b/utils.py
index 456fbc8..42265d3 100644
--- a/utils.py
+++ b/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
\ No newline at end of file
+ #print(categories_array_filtered)
+ #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
--
GitLab