lhmd commited on
Commit
cfea088
·
verified ·
1 Parent(s): 1e76d3f

Update convert_dl3dv_train.py

Browse files
Files changed (1) hide show
  1. convert_dl3dv_train.py +1 -1
convert_dl3dv_train.py CHANGED
@@ -142,7 +142,7 @@ def is_image_shape_matched(image_dir, target_shape):
142
 
143
  def legal_check_for_all_scenes(root_dir, target_shape):
144
  valid_folders = []
145
- sub_folders = sorted(glob(os.path.join(root_dir, "*/*")))
146
  for sub_folder in tqdm(sub_folders, desc="checking scenes..."):
147
  # img_dir = os.path.join(sub_folder, 'images_8')
148
  img_dir = os.path.join(sub_folder, "images_4")
 
142
 
143
  def legal_check_for_all_scenes(root_dir, target_shape):
144
  valid_folders = []
145
+ sub_folders = sorted(glob(os.path.join(root_dir, "*")))
146
  for sub_folder in tqdm(sub_folders, desc="checking scenes..."):
147
  # img_dir = os.path.join(sub_folder, 'images_8')
148
  img_dir = os.path.join(sub_folder, "images_4")