lhmd commited on
Commit
6a697f6
·
verified ·
1 Parent(s): 0b58238

Update convert_dl3dv_train.py

Browse files
Files changed (1) hide show
  1. convert_dl3dv_train.py +2 -2
convert_dl3dv_train.py CHANGED
@@ -143,8 +143,8 @@ def legal_check_for_all_scenes(root_dir, target_shape):
143
  valid_folders = []
144
  sub_folders = sorted(glob(os.path.join(root_dir, "*/*")))
145
  for sub_folder in tqdm(sub_folders, desc="checking scenes..."):
146
- img_dir = os.path.join(sub_folder, 'images_8')
147
- # img_dir = os.path.join(sub_folder, "images_4")
148
  if not is_image_shape_matched(Path(img_dir), target_shape):
149
  print(f"image shape does not match for {sub_folder}")
150
  continue
 
143
  valid_folders = []
144
  sub_folders = sorted(glob(os.path.join(root_dir, "*/*")))
145
  for sub_folder in tqdm(sub_folders, desc="checking scenes..."):
146
+ # img_dir = os.path.join(sub_folder, 'images_8')
147
+ img_dir = os.path.join(sub_folder, "images_4")
148
  if not is_image_shape_matched(Path(img_dir), target_shape):
149
  print(f"image shape does not match for {sub_folder}")
150
  continue