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

Merge branch 'master' of https://git.hofee.top/hofee/nbv_reconstruction

Browse Source
This commit is contained in:
hofee 2024-10-05 15:11:01 -05:00
commit 11d460bd9b
2 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File 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: 5002 port: 5004
from: 2200 from: 4000
to: 2300 # -1 means all to: -1 # -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

6
preprocess/preprocessor.py
View File

@ -78,7 +78,11 @@ 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(
@ -148,6 +152,8 @@ 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