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

Update convert_dl3dv_train.py

Browse files
Files changed (1) hide show
  1. convert_dl3dv_train.py +5 -0
convert_dl3dv_train.py CHANGED
@@ -215,6 +215,11 @@ def process_single_directory(input_dir: Path, output_dir: Path):
215
 
216
  image_dir = Path(image_dir) / 'images_4' # 540x960
217
 
 
 
 
 
 
218
  num_bytes = get_size(image_dir)
219
 
220
  # Read images and metadata.
 
215
 
216
  image_dir = Path(image_dir) / 'images_4' # 540x960
217
 
218
+ # Check if image directory exists
219
+ if not image_dir.exists():
220
+ print(f"Image directory does not exist for {key}, skipping...")
221
+ continue
222
+
223
  num_bytes = get_size(image_dir)
224
 
225
  # Read images and metadata.