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-17 11:11:14 +00:00
commit 22e7a1aed4
3 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
configs/local/view_generate_config.yaml
View File

@ -8,11 +8,11 @@ runner:
root_dir: experiments root_dir: experiments
generate: generate:
port: 5004 port: 5004
from: 590 from: 0
to: 2000 # -1 means all to: 1 # -1 means all
object_dir: /media/hofee/data/data/scaled_object_meshes object_dir: /media/hofee/data/data/box_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/full_data_output output_dir: /media/hofee/data/data/box_views
binocular_vision: true binocular_vision: true
plane_size: 10 plane_size: 10
max_views: 512 max_views: 512

3
preprocess/preprocessor.py
View File

@ -9,12 +9,15 @@ from utils.reconstruction import ReconstructionUtil
from utils.data_load import DataLoadUtil from utils.data_load import DataLoadUtil
from utils.pts import PtsUtil from utils.pts import PtsUtil
# scan shoe 536
def save_np_pts(path, pts: np.ndarray, file_type="txt"): def save_np_pts(path, pts: np.ndarray, file_type="txt"):
if file_type == "txt": if file_type == "txt":
np.savetxt(path, pts) np.savetxt(path, pts)
else: else:
np.save(path, pts) np.save(path, pts)
def save_target_points(root, scene, frame_idx, target_points: np.ndarray, file_type="txt"): def save_target_points(root, scene, frame_idx, target_points: np.ndarray, file_type="txt"):
pts_path = os.path.join(root,scene, "pts", f"{frame_idx}.{file_type}") pts_path = os.path.join(root,scene, "pts", f"{frame_idx}.{file_type}")
if not os.path.exists(os.path.join(root,scene, "pts")): if not os.path.exists(os.path.join(root,scene, "pts")):

3
runners/view_generator.py
View File

@ -9,7 +9,8 @@ class ViewGenerator(Runner):
self.config_path = config_path self.config_path = config_path
def run(self): def run(self):
subprocess.run(['blender', '-b', '-P', '../blender/run_blender.py', '--', self.config_path]) result = subprocess.run(['blender', '-b', '-P', '../blender/run_blender.py', '--', self.config_path])
print()
def create_experiment(self, backup_name=None): def create_experiment(self, backup_name=None):
return super().create_experiment(backup_name) return super().create_experiment(backup_name)