hofee/nbv_reconstruction
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add from to in generating strategy

Browse Source
This commit is contained in:
hofee 2024-09-30 10:04:53 +08:00
parent cb983fdc74
commit 983cb22d4c
3 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
configs/local/strategy_generate_config.yaml
View File

@ -27,8 +27,10 @@ runner:
datasets: datasets:
OmniObject3d: OmniObject3d:
#"/media/hofee/data/data/temp_output" #"/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" model_dir: "/media/hofee/data/data/scaled_object_meshes"
from: 0
to: -1 # -1 means all
#output_dir: "/media/hofee/data/data/label_output" #output_dir: "/media/hofee/data/data/label_output"

4
configs/local/view_generate_config.yaml
View File

@ -12,8 +12,8 @@ runner:
output_dir: /media/hofee/repository/new_nbv_reconstruction_data_512 output_dir: /media/hofee/repository/new_nbv_reconstruction_data_512
binocular_vision: true binocular_vision: true
plane_size: 10 plane_size: 10
max_views: 120 max_views: 512
min_views: 60 min_views: 64
random_view_ratio: 0.2 random_view_ratio: 0.2
min_cam_table_included_degree: 20 min_cam_table_included_degree: 20
max_diag: 0.7 max_diag: 0.7

7
runners/strategy_generator.py
View File

@ -41,10 +41,15 @@ class StrategyGenerator(Runner):
status_manager.set_progress("generate_strategy", "strategy_generator", "dataset", dataset_idx, len(dataset_name_list)) status_manager.set_progress("generate_strategy", "strategy_generator", "dataset", dataset_idx, len(dataset_name_list))
root_dir = ConfigManager.get("datasets", dataset_name, "root_dir") root_dir = ConfigManager.get("datasets", dataset_name, "root_dir")
model_dir = ConfigManager.get("datasets", dataset_name, "model_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) scene_name_list = os.listdir(root_dir)
if to_idx == -1:
to_idx = len(scene_name_list)
cnt = 0 cnt = 0
total = len(scene_name_list) 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}") Log.info(f"({dataset_name})Processing [{cnt}/{total}]: {scene_name}")
status_manager.set_progress("generate_strategy", "strategy_generator", "scene", cnt, total) status_manager.set_progress("generate_strategy", "strategy_generator", "scene", cnt, total)
diag = DataLoadUtil.get_bbox_diag(model_dir, scene_name) diag = DataLoadUtil.get_bbox_diag(model_dir, scene_name)