Update convert_dl3dv_train.py
Browse files- 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.
|