Skip to content
Snippets Groups Projects
Commit 6664440e authored by Guillaume Duret's avatar Guillaume Duret
Browse files

Merge branch 'dev' of https://gitlab.liris.cnrs.fr/maali/guimod into dev

parents b9501e11 b15e8a39
No related branches found
No related tags found
No related merge requests found
...@@ -59,6 +59,11 @@ def process_compute(data_name, camera, Nb_camera, World_begin, Nb_world, list_ca ...@@ -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) : if len(data_Bbox_2d) != len(data_3D_pose) :
raise TypeError("size of datas are differents !!") 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: for categories in list_categories:
...@@ -156,6 +161,8 @@ def process_compute(data_name, camera, Nb_camera, World_begin, Nb_world, list_ca ...@@ -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) np.savetxt(f'{data_name}/Generated/FPS/{categories}/{p}.txt', out)
#print("stop") #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) print(list_count_categories)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment