complete splitable script

compute_features.py

@@ -59,6 +59,11 @@ def process_compute(data_name, camera, Nb_camera, World_begin, Nb_world, list_ca
 
             if len(data_Bbox_2d) != len(data_3D_pose) :
                 raise TypeError("size of datas are differents !!")
+            
+            if os.path.isfile(f'{data_name}/Generated/Count_{p-1}.json'):
+                with open(f'{data_name}/Generated/Count_{p-1}.json') as f:
+                    list_count_categories = json.load(f)
+
 
             for categories in list_categories:
 
@@ -156,6 +161,8 @@ def process_compute(data_name, camera, Nb_camera, World_begin, Nb_world, list_ca
                         np.savetxt(f'{data_name}/Generated/FPS/{categories}/{p}.txt', out)
                         #print("stop")
 
+    with open(f'{data_name}/Generated/Count_{p}.json', mode='w') as f:
+        f.write(json.dumps(list_count_categories, indent=4))
     print(list_count_categories)