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

updaaaaaaaaaaate

Browse Source
This commit is contained in:
hofee 2024-10-11 16:34:16 +08:00
parent d564701807
commit dee7211e0b
3 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
configs/local/strategy_generate_config.yaml
View File

@ -11,7 +11,7 @@ runner:
root_dir: "experiments" root_dir: "experiments"
generate: generate:
voxel_threshold: 0.01 voxel_threshold: 0.003
soft_overlap_threshold: 0.3 soft_overlap_threshold: 0.3
hard_overlap_threshold: 0.6 hard_overlap_threshold: 0.6
filter_degree: 75 filter_degree: 75
@ -21,15 +21,15 @@ runner:
save_best_combined_points: False save_best_combined_points: False
save_mesh: True save_mesh: True
overwrite: False overwrite: False
seq_num: 10 seq_num: 15
dataset_list: dataset_list:
- OmniObject3d - OmniObject3d
datasets: datasets:
OmniObject3d: OmniObject3d:
#"/media/hofee/data/data/temp_output" #"/media/hofee/data/data/temp_output"
root_dir: "/media/hofee/data/data/box_output" root_dir: /media/hofee/repository/full_data_output
model_dir: "/media/hofee/data/data/box_object_meshes" model_dir: /media/hofee/data/data/scaled_object_meshes
from: 0 from: 0
to: -1 # -1 means end to: -1 # -1 means end
#output_dir: "/media/hofee/data/data/label_output" #output_dir: "/media/hofee/data/data/label_output"

8
preprocess/preprocessor.py
View File

@ -67,7 +67,7 @@ def save_scene_data(root, scene, scene_idx=0, scene_total=1,file_type="txt"):
target_mask_label = (0, 255, 0, 255) target_mask_label = (0, 255, 0, 255)
display_table_mask_label=(0, 0, 255, 255) display_table_mask_label=(0, 0, 255, 255)
random_downsample_N = 32768 random_downsample_N = 32768
voxel_size=0.002 voxel_size=0.003
filter_degree = 75 filter_degree = 75
min_z = 0.2 min_z = 0.2
max_z = 0.5 max_z = 0.5
@ -135,7 +135,7 @@ def save_scene_data(root, scene, scene_idx=0, scene_total=1,file_type="txt"):
if __name__ == "__main__": if __name__ == "__main__":
#root = "/media/hofee/repository/new_data_with_normal" #root = "/media/hofee/repository/new_data_with_normal"
root = r"/media/hofee/data/data/box_output" root = r"/media/hofee/repository/full_data_output"
# list_path = r"/media/hofee/repository/full_list.txt" # list_path = r"/media/hofee/repository/full_list.txt"
# scene_list = [] # scene_list = []
@ -144,7 +144,7 @@ if __name__ == "__main__":
# scene_list.append(line.strip()) # scene_list.append(line.strip())
scene_list = os.listdir(root) scene_list = os.listdir(root)
from_idx = 0 # 1000 from_idx = 0 # 1000
to_idx = len(scene_list) # 1500 to_idx = 700 # 1500
print(scene_list) print(scene_list)
@ -153,7 +153,7 @@ if __name__ == "__main__":
total = to_idx - from_idx total = to_idx - from_idx
for scene in scene_list[from_idx:to_idx]: for scene in scene_list[from_idx:to_idx]:
start = time.time() start = time.time()
save_scene_data(root, scene, cnt, total, file_type="txt") save_scene_data(root, scene, cnt, total, file_type="npy")
cnt+=1 cnt+=1
end = time.time() end = time.time()
print(f"Time cost: {end-start}") print(f"Time cost: {end-start}")

2
runners/strategy_generator.py
View File

@ -102,7 +102,7 @@ class StrategyGenerator(Runner):
seq_num = min(self.seq_num, non_zero_cnt) seq_num = min(self.seq_num, non_zero_cnt)
init_view_list = [] init_view_list = []
idx = 0 idx = 0
while len(init_view_list) < seq_num: while len(init_view_list) < seq_num and idx < len(pts_list):
if pts_list[idx].shape[0] > 100: if pts_list[idx].shape[0] > 100:
init_view_list.append(idx) init_view_list.append(idx)
idx += 1 idx += 1