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

Update convert_dl3dv_train.py

Browse files
Files changed (1) hide show
  1. convert_dl3dv_train.py +2 -3
convert_dl3dv_train.py CHANGED
@@ -22,7 +22,7 @@ parser.add_argument("--output_base_dir", type=str, help="base output directory f
22
  parser.add_argument(
23
  "--img_subdir",
24
  type=str,
25
- default="images_8",
26
  help="image directory name",
27
  choices=[
28
  "images_4",
@@ -213,8 +213,7 @@ def process_single_directory(input_dir: Path, output_dir: Path):
213
  print(f"scene {key} in benchmark, skip.")
214
  continue
215
 
216
- image_dir = Path(image_dir) / "images_8" # 270x480
217
- # image_dir = Path(image_dir) / 'images_4' # 540x960
218
 
219
  num_bytes = get_size(image_dir)
220
 
 
22
  parser.add_argument(
23
  "--img_subdir",
24
  type=str,
25
+ default="images_4",
26
  help="image directory name",
27
  choices=[
28
  "images_4",
 
213
  print(f"scene {key} in benchmark, skip.")
214
  continue
215
 
216
+ image_dir = Path(image_dir) / 'images_4' # 540x960
 
217
 
218
  num_bytes = get_size(image_dir)
219