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

Compare commits

base: hofee:dc79f4b31301ffe8a45aede0e6302e1393642c8a
Branches Tags
hofee:master
hofee:ab_new_partial_global
hofee:new_partial_global
hofee:ab_global_only
hofee:hemisphere_random
hofee:ab_global_and_partial_global
hofee:ab_global_and_local
hofee:ab_local_only
..
compare: hofee:d098c9f9518fdc477b3feb916e749f2e060e0bac
Branches Tags
hofee:ab_new_partial_global
hofee:new_partial_global
hofee:ab_global_only
hofee:hemisphere_random
hofee:master
hofee:ab_global_and_partial_global
hofee:ab_global_and_local
hofee:ab_local_only

No commits in common. "dc79f4b31301ffe8a45aede0e6302e1393642c8a" and "d098c9f9518fdc477b3feb916e749f2e060e0bac" have entirely different histories.
dc79f4b313 ... d098c9f951

2 changed files with 6 additions and 12 deletions
Show Stats Expand all files Collapse all files

6
configs/local/view_generate_config.yaml
View File

@ -7,9 +7,9 @@ runner:
name: debug name: debug
root_dir: experiments root_dir: experiments
generate: generate:
port: 5004 port: 5002
from: 4000 from: 2200
to: -1 # -1 means all to: 2300 # -1 means all
object_dir: /media/hofee/data/data/scaled_object_meshes object_dir: /media/hofee/data/data/scaled_object_meshes
table_model_path: /media/hofee/data/data/others/table.obj table_model_path: /media/hofee/data/data/others/table.obj
output_dir: /media/hofee/repository/new_data_with_normal output_dir: /media/hofee/repository/new_data_with_normal

8
preprocess/preprocessor.py
View File

@ -74,7 +74,7 @@ def get_scan_points_indices(scan_points, mask, display_table_mask_label, cam_int
return selected_points_indices return selected_points_indices
def save_scene_data(root, scene, scene_idx=0, scene_total=1,file_type="txt"): def save_scene_data(root, scene, scene_idx=0, scene_total=1):
''' configuration ''' ''' configuration '''
target_mask_label = (0, 255, 0, 255) target_mask_label = (0, 255, 0, 255)
@ -94,11 +94,7 @@ def save_scene_data(root, scene, scene_idx=0, scene_total=1,file_type="txt"):
''' read frame data(depth|mask|normal) ''' ''' read frame data(depth|mask|normal) '''
frame_num = DataLoadUtil.get_scene_seq_length(root, scene) frame_num = DataLoadUtil.get_scene_seq_length(root, scene)
for frame_id in range(frame_num): for frame_id in range(frame_num):
print(f"[scene({scene_idx}/{scene_total})|frame({frame_id}/{frame_num})]Processing {scene} frame {frame_id}") print(f"[scene({scene_idx}/{scene_total})|frame({frame_id}/{frame_num})]Processing {scene} frame {frame_id}")
if frame_id != 126:
continue
path = DataLoadUtil.get_path(root, scene, frame_id) path = DataLoadUtil.get_path(root, scene, frame_id)
cam_info = DataLoadUtil.load_cam_info(path, binocular=True) cam_info = DataLoadUtil.load_cam_info(path, binocular=True)
depth_L, depth_R = DataLoadUtil.load_depth( depth_L, depth_R = DataLoadUtil.load_depth(
@ -167,8 +163,6 @@ if __name__ == "__main__":
from_idx = 0 from_idx = 0
to_idx = len(scene_list) to_idx = len(scene_list)
cnt = 0 cnt = 0
import time import time
total = to_idx - from_idx total = to_idx - from_idx