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

add min_cam_table_included_degree and random_view

Browse Source
This commit is contained in:
hofee
2024-09-28 22:03:19 +08:00
parent 70280a7b92
commit f94cb2c0d6
3 changed files with 170 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
data_generator.py
View File

@@ -21,6 +21,8 @@ class DataGenerator:
self.min_views = config["runner"]["generate"]["min_views"]
self.min_diag = config["runner"]["generate"]["min_diag"]
self.max_diag = config["runner"]["generate"]["max_diag"]
self.min_cam_table_included_degree = config["runner"]["generate"]["min_cam_table_included_degree"]
self.random_view_ratio = config["runner"]["generate"]["random_view_ratio"]
self.binocular_vision = config["runner"]["generate"]["binocular_vision"]
self.set_status_path = "http://localhost:5000/project/set_status"
self.log_path = "http://localhost:5000/project/add_log"
@@ -240,7 +242,7 @@ class DataGenerator:
if not os.path.exists(scene_dir):
os.makedirs(scene_dir)
view_num = int(self.min_views + (diag - self.min_diag)/(self.max_diag - self.min_diag) * (self.max_views - self.min_views))
view_data = ViewSampleUtil.sample_view_data_world_space(self.target_obj, distance_range=(0.2,0.4), voxel_size=0.005, max_views=view_num)
view_data = ViewSampleUtil.sample_view_data_world_space(self.target_obj, distance_range=(0.2,0.4), voxel_size=0.005, max_views=view_num, min_cam_table_included_degree = self.min_cam_table_included_degree, random_view_ratio = self.random_view_ratio )
object_points = np.array(view_data["voxel_down_sampled_points"])
normals = np.array(view_data["normals"])
points_normals = np.concatenate((object_points, normals), axis=1)