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