From d42ec3773d890f8d3f45248c104f49d65660d938 Mon Sep 17 00:00:00 2001 From: Gduret <guillaume.duret@ec-lyon.fr> Date: Sat, 25 Feb 2023 15:29:45 +0100 Subject: [PATCH] choose how many world to compute --- compute_features.py | 4 ++-- main.py | 17 ++++++++++++++--- prepare_data.py | 4 ++-- 3 files changed, 18 insertions(+), 7 deletions(-) diff --git a/compute_features.py b/compute_features.py index 72894b8..7ca2537 100644 --- a/compute_features.py +++ b/compute_features.py @@ -31,13 +31,13 @@ def updateJsonFile(): jsonFile.close() -def process_compute(data_name, camera, Nb_camera, Nb_world, list_categories, occ_target, vis): +def process_compute(data_name, camera, 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], [1.0000000, 0.0000000, 0.0000000]]) list_count_categories = {} - for i in range(1, Nb_world + 1): # worlds + for i in range(World_begin, World_begin + Nb_world): # worlds catergories_instance_array_id_to_cat, catergories_instance_array_cat_to_id, catergories_label_to_id = compute_categories_id(data_name, i) diff --git a/main.py b/main.py index 19a211d..365fd3c 100644 --- a/main.py +++ b/main.py @@ -10,6 +10,8 @@ from bbox_3d import generate_3d_bbox from compute_features import process_compute import shutil import open3d as o3d +# Import the library +import argparse def generate_folders(name, list_categories): is_exist = os.path.exists(name) @@ -27,6 +29,15 @@ def generate_folders(name, list_categories): if __name__ == '__main__': + # Create the parser + parser = argparse.ArgumentParser() + # Add an argument + parser.add_argument('--Nb_worlds', type=int, required=True) + parser.add_argument('--World_begin', type=int, required=True) + # Parse the argument + args = parser.parse_args() + + ### parameters ### Categories = [] # to read Nb_instance = 1 @@ -42,7 +53,7 @@ if __name__ == '__main__': # frame = "1_600000000" frame = "1_926000000" Nb_camera = 15 - Nb_world = 2 + #Nb_world = 2 generate_folders(dataset_name, list_categories) for cat in list_categories: @@ -72,9 +83,9 @@ if __name__ == '__main__': np.savetxt(f'{dataset_name}/Generated/camera_{choice}.txt', camera) - reform_data(dataset_src, dataset_name, dataset_type, frame, Nb_camera, Nb_world) + reform_data(dataset_src, dataset_name, dataset_type, frame, Nb_camera, args.World_begin, args.Nb_worlds) - process_compute(dataset_name, camera, Nb_camera, Nb_world, list_categories, occ_target, False) + process_compute(dataset_name, camera, Nb_camera, args.World_begin, args.Nb_worlds, list_categories, occ_target, True) #transform_pose(dataset_name, Nb_camera, Nb_world, list_categories, occ_target) #generate_2d_bbox(dataset_name, Nb_camera, Nb_world, list_categories, occ_target) #generate_instance_mask(dataset_name, Nb_camera, Nb_world, list_categories, occ_target) diff --git a/prepare_data.py b/prepare_data.py index 42ec580..667436f 100644 --- a/prepare_data.py +++ b/prepare_data.py @@ -1,8 +1,8 @@ import shutil -def reform_data(src, data_name, data_option, frame, Nb_camera, Nb_world): - for i in range(1, Nb_world + 1): # worlds +def reform_data(src, data_name, data_option, frame, Nb_camera, World_begin, Nb_world): + for i in range(World_begin, World_begin + Nb_world): # worlds src_meta = f"{src}/{i}/meta.json" dst_meta = f"{data_name}/Meta/{i}.json" shutil.copy(src_meta, dst_meta) -- GitLab