diff --git a/runners/strategy_generator.py b/runners/strategy_generator.py index 2a24875..399b2ae 100644 --- a/runners/strategy_generator.py +++ b/runners/strategy_generator.py @@ -40,7 +40,6 @@ class StrategyGenerator(Runner): dataset_name = dataset_name_list[dataset_idx] 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) @@ -52,17 +51,13 @@ class StrategyGenerator(Runner): 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) - # status_manager.set_status("generate_strategy", "strategy_generator", "diagonal", diag) output_label_path = DataLoadUtil.get_label_path(root_dir, scene_name,0) if os.path.exists(output_label_path) and not self.overwrite: Log.info(f"Scene <{scene_name}> Already Exists, Skip") cnt += 1 continue - self.generate_sequence(root_dir, None, scene_name,voxel_threshold, soft_overlap_threshold, hard_overlap_threshold) - # except Exception as e: - # Log.error(f"Scene <{scene_name}> Failed, Error: {e}") + self.generate_sequence(root_dir, scene_name,voxel_threshold, soft_overlap_threshold, hard_overlap_threshold) cnt += 1 status_manager.set_progress("generate_strategy", "strategy_generator", "scene", total, total) status_manager.set_progress("generate_strategy", "strategy_generator", "dataset", len(dataset_name_list), len(dataset_name_list)) @@ -75,7 +70,7 @@ class StrategyGenerator(Runner): def load_experiment(self, backup_name=None): super().load_experiment(backup_name) - def generate_sequence(self, root, model_dir, scene_name, voxel_threshold, soft_overlap_threshold, hard_overlap_threshold): + def generate_sequence(self, root, scene_name, voxel_threshold, soft_overlap_threshold, hard_overlap_threshold): status_manager.set_status("generate_strategy", "strategy_generator", "scene", scene_name) frame_num = DataLoadUtil.get_scene_seq_length(root, scene_name) import ipdb; ipdb.set_trace() @@ -133,8 +128,6 @@ class StrategyGenerator(Runner): with open(output_label_path, 'w') as f: json.dump(seq_save_data, f) seq_idx += 1 - # if self.save_mesh: - # DataLoadUtil.save_target_mesh_at_world_space(root, model_dir, scene_name) status_manager.set_progress("generate_strategy", "strategy_generator", "computing sequence", len(init_view_list), len(init_view_list))