From b9501e11dc38cbb736fce76804b24ae2f2f4e392 Mon Sep 17 00:00:00 2001
From: Guillaume-Duret <guillaume.duret@ec-lyon.fr>
Date: Sun, 26 Feb 2023 01:02:11 +0100
Subject: [PATCH] ready for jz ?
---
main.py | 12 +++++-------
1 file changed, 5 insertions(+), 7 deletions(-)
diff --git a/main.py b/main.py
index 9684bb7..cced98f 100644
--- a/main.py
+++ b/main.py
@@ -29,8 +29,7 @@ def generate_folders(name, list_categories):
if not is_exist2:
os.makedirs(f"{name}/Generated/{f}/{cat}")
-if __name__ == '__main__':
-
+if __name__ == '__main__':
# Create the parser
parser = argparse.ArgumentParser()
# Add an argument
@@ -39,23 +38,22 @@ if __name__ == '__main__':
# Parse the argument
args = parser.parse_args()
-
### parameters ###
Categories = [] # to read
Nb_instance = 1
occ_target = 0.5
- dataset_src = "/media/gduret/DATA/dataset/s2rg/Fruits_all_medium/data"
+ dataset_src = "/gpfsscratch/rech/uli/ubn15wo/dataset/s2rg/Fruits_all_medium/data/"
#dataset_src = "/media/mahmoud/E/Fruits_easy/data"
choice = "low" # depth of rgb resolution datas
data_options = {"high": "ground_truth_rgb",
"low": "ground_truth_depth"}
dataset_type = data_options[choice]
- dataset_name = f"GUIMOD_{choice}"
+ dataset_name = f"/gpfsscratch/rech/uli/ubn15wo/dataset/s2rg/Fruits_all_medium/GUIMOD_{choice}"
list_categories = ["banana1", "kiwi1", "pear2", "strawberry1", "apricot", "orange2", "peach1", "lemon2", "apple2" ]
# frame = "1_600000000"
#frame = "1_926000000"
Nb_camera = 15
- #Nb_world = 2
+ #Nb_world = 10000
generate_folders(dataset_name, list_categories)
for cat in list_categories:
@@ -69,7 +67,7 @@ if __name__ == '__main__':
#print("pcd", pcd)
- fps_points = apply_fps(pcd, 200)
+ fps_points = apply_fps(pcd, 8)
#print(fps_points)
np.savetxt(f'{dataset_name}/Generated/FPS/{categories}_fps_3d.txt', fps_points)
--
GitLab