Update convert_dl3dv_train.py
Browse files- 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="
|
| 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) /
|
| 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 |
|