diff --git a/configs/local/strategy_generate_config.yaml b/configs/local/strategy_generate_config.yaml index 5bcd80a..3ef8221 100644 --- a/configs/local/strategy_generate_config.yaml +++ b/configs/local/strategy_generate_config.yaml @@ -27,8 +27,10 @@ runner: datasets: OmniObject3d: #"/media/hofee/data/data/temp_output" - root_dir: "/media/hofee/data/data/sample_data/view_data" + root_dir: "/media/hofee/repository/new_sample" model_dir: "/media/hofee/data/data/scaled_object_meshes" + from: 0 + to: -1 # -1 means all #output_dir: "/media/hofee/data/data/label_output" diff --git a/configs/local/view_generate_config.yaml b/configs/local/view_generate_config.yaml index 624adb7..09a5da8 100644 --- a/configs/local/view_generate_config.yaml +++ b/configs/local/view_generate_config.yaml @@ -12,8 +12,8 @@ runner: output_dir: /media/hofee/repository/new_nbv_reconstruction_data_512 binocular_vision: true plane_size: 10 - max_views: 120 - min_views: 60 + max_views: 512 + min_views: 64 random_view_ratio: 0.2 min_cam_table_included_degree: 20 max_diag: 0.7 diff --git a/runners/strategy_generator.py b/runners/strategy_generator.py index d6c1598..c8f91be 100644 --- a/runners/strategy_generator.py +++ b/runners/strategy_generator.py @@ -41,10 +41,15 @@ class StrategyGenerator(Runner): status_manager.set_progress("generate_strategy", "strategy_generator", "dataset", dataset_idx, len(dataset_name_list)) root_dir = ConfigManager.get("datasets", dataset_name, "root_dir") model_dir = ConfigManager.get("datasets", dataset_name, "model_dir") + from_idx = ConfigManager.get("datasets",dataset_name,"from") + to_idx = ConfigManager.get("datasets",dataset_name,"to") scene_name_list = os.listdir(root_dir) + if to_idx == -1: + to_idx = len(scene_name_list) cnt = 0 total = len(scene_name_list) - for scene_name in scene_name_list: + Log.info(f"Processing Dataset: {dataset_name}, From: {from_idx}, To: {to_idx}") + for scene_name in scene_name_list[from_idx:to_idx]: Log.info(f"({dataset_name})Processing [{cnt}/{total}]: {scene_name}") status_manager.set_progress("generate_strategy", "strategy_generator", "scene", cnt, total) diag = DataLoadUtil.get_bbox_diag(model_dir, scene_name)