compute statistics

compute_features.py

@@ -12,7 +12,6 @@ from fps_alg import process2
 import os
 
 
-
 def process_compute(data_name, camera, camera_resized, new_size, Nb_camera, World_begin, Nb_world, list_categories, occ_target, vis):
     transformation = np.matrix([[0.0000000, -1.0000000, 0.0000000],
                                 [0.0000000, 0.0000000, -1.0000000],
@@ -50,36 +49,43 @@ def process_compute(data_name, camera, camera_resized, new_size, Nb_camera, Worl
 
             for categories in list_categories:
                 if categories in catergories_occ_array.keys():
-                    if len(catergories_occ_array[categories]) == 1 :
-
-                        meta = {}
-
-                        if categories in list_count_categories.keys():
-                            list_count_categories[categories] += 1
-                        else:
-                            list_count_categories[categories] = 1
-
-                        meta['id_generated'] = list_count_categories[categories] 
-                        meta['id_original'] = p 
-                        meta['id_category'] = catergories_label_to_id[categories] 
-                        meta['id_instance'] = catergories_occ_array[categories][0] 
-                        meta['id_dataset'] = 1
-                        meta['world'] = i
-                        meta['camera'] = f"grabber_{j}"
-                        meta['occlusion'] = occ_target
-                        meta['Nb_instance_category'] = 1
-
-                        if not os.path.isfile(f'{data_name}/Generated/{categories}/Meta_Gen/{categories}.json'):
-                            with open(f'{data_name}/Generated/{categories}/Meta_Gen/{categories}.json', mode='w') as f:
-                                feeds = {}
-                                feeds[meta['id_generated']]=meta
-                                f.write(json.dumps(feeds, indent=2))
-                        else:
-                            with open(f'{data_name}/Generated/{categories}/Meta_Gen/{categories}.json') as feedsjson:
-                                feeds = json.load(feedsjson)
+                    Nb_instance = len(catergories_occ_array[categories])
+
+                    meta = {}
+                    if categories in list_count_categories.keys():
+                        print("hello")
+                    else:
+                        #list_count_categories[categories] = {categories}
+                        list_count_categories[categories] = {}
+
+                    if Nb_instance in list_count_categories[categories].keys() : 
+                        list_count_categories[categories][Nb_instance] += 1
+                    else : 
+                        list_count_categories[categories][Nb_instance] = 1
+
+                    meta['id_generated'] = list_count_categories[categories][Nb_instance]
+                    meta['id_original'] = p
+                    meta['id_category'] = catergories_label_to_id[categories] 
+                    meta['id_instance'] = catergories_occ_array[categories]
+                    meta['id_dataset'] = 1
+                    meta['world'] = i
+                    meta['camera'] = f"grabber_{j}"
+                    meta['occlusion'] = occ_target
+                    meta['Nb_instance_category'] = Nb_instance
+
+                    if not os.path.isfile(f'{data_name}/Generated/{categories}/Meta_Gen/{categories}.json'):
+                        with open(f'{data_name}/Generated/{categories}/Meta_Gen/{categories}.json', mode='w') as f:
+                            feeds = {}
+                            feeds[meta['id_generated']]=meta
+                            f.write(json.dumps(feeds, indent=2))
+                    else:
+                        with open(f'{data_name}/Generated/{categories}/Meta_Gen/{categories}.json') as feedsjson:
+                            feeds = json.load(feedsjson)
                             feeds[meta['id_generated']]=meta
-                            with open(f'{data_name}/Generated/{categories}/Meta_Gen/{categories}.json', mode='w') as f:
-                                f.write(json.dumps(feeds, indent=4))
+                        with open(f'{data_name}/Generated/{categories}/Meta_Gen/{categories}.json', mode='w') as f:
+                            f.write(json.dumps(feeds, indent=4))    
+
+                    if (Nb_instance == 1):
 
                         for k in range(len(data_3D_pose)):