udpate data_preparing.py

Lin Shao · Lin Shao · commit df94e274c153 · 2020-09-14T16:05:39.000-07:00
diff --git a/point_set_selection/data_preparing.py b/point_set_selection/data_preparing.py
@@ -11,13 +11,13 @@
 
 for line_id in os.listdir(data_top_dir):
   sub_dir = os.path.join(data_top_dir,line_id)
-  #print(sub_dir)
+  print(sub_dir)
   #f1 = [line for line in os.listdir(sub_dir) if line.endswith('grasp1_new.npz')]
   #f2 = [line for line in os.listdir(sub_dir) if line.endswith('grasp2_new.npz')]
   #f3 = [line for line in os.listdir(sub_dir) if line.endswith('grasp3_new.npz')]
-  #f4 = [line for line in os.listdir(sub_dir) if line.endswith('grasp4_new.npz')]
+  f4 = [line for line in os.listdir(sub_dir) if line.endswith('grasp4_new.npz')]
   #f5 = [line for line in os.listdir(sub_dir) if line.endswith('grasp5_new.npz')]
-  if 1:#len(f4) > 0 and len(f2) > 0 and len(f3) > 0 and len(f1) > 0 and len(f5) > 0:
+  if len(f4) > 0:# and len(f2) > 0 and len(f3) > 0 and len(f1) > 0 and len(f5) > 0:
     total_ids.append(line_id)
 
 train_val_test_list = Train_Val_Test(np.array(total_ids),splitting=[1.0,0.0,0.0])
