diff --git a/compute_features.py b/compute_features.py index 7ca253729179f11351753dba1f03cbc7e91207c0..c048b49970da3da9354a9ee88fedf3ac30468e87 100644 --- a/compute_features.py +++ b/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)