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

solve merge conflicts

Browse Source
This commit is contained in:
hofee 2024-10-05 15:12:55 -05:00
commit 60c9357491
1 changed files with 4 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
preprocess/preprocessor.py
View File

@ -78,11 +78,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(
@ -142,16 +138,16 @@ 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"C:\Document\Local Project\nbv_rec\nbv_reconstruction\test\test_sample" root = r"/media/hofee/repository/new_data_with_normal"
list_path = r"C:\Document\Local Project\nbv_rec\nbv_reconstruction\test\test_sample/test_sample_list.txt" list_path = r"/media/hofee/repository/full_list.txt"
scene_list = [] scene_list = []
with open(list_path, "r") as f: with open(list_path, "r") as f:
for line in f: for line in f:
scene_list.append(line.strip()) scene_list.append(line.strip())
from_idx = 0 from_idx = 1000
to_idx = len(scene_list) to_idx = 1500
cnt = 0 cnt = 0