configs/local/view_generate_config.yaml

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

preprocess/preprocessor.py

@@ -74,7 +74,7 @@ def get_scan_points_indices(scan_points, mask, display_table_mask_label, cam_int
     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 '''
     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) '''
     frame_num = DataLoadUtil.get_scene_seq_length(root, scene)
     for frame_id in range(frame_num):
-    
-        
         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)
         cam_info = DataLoadUtil.load_cam_info(path, binocular=True)
         depth_L, depth_R = DataLoadUtil.load_depth(
@@ -167,8 +163,6 @@ if __name__ == "__main__":
 
     from_idx = 0
     to_idx = len(scene_list)
-
-
     cnt = 0
     import time
     total = to_idx - from_idx