Update convert_dl3dv_train.py
Browse files- convert_dl3dv_train.py +1 -1
convert_dl3dv_train.py
CHANGED
|
@@ -142,7 +142,7 @@ def is_image_shape_matched(image_dir, target_shape):
|
|
| 142 |
|
| 143 |
def legal_check_for_all_scenes(root_dir, target_shape):
|
| 144 |
valid_folders = []
|
| 145 |
-
sub_folders = sorted(glob(os.path.join(root_dir, "
|
| 146 |
for sub_folder in tqdm(sub_folders, desc="checking scenes..."):
|
| 147 |
# img_dir = os.path.join(sub_folder, 'images_8')
|
| 148 |
img_dir = os.path.join(sub_folder, "images_4")
|
|
|
|
| 142 |
|
| 143 |
def legal_check_for_all_scenes(root_dir, target_shape):
|
| 144 |
valid_folders = []
|
| 145 |
+
sub_folders = sorted(glob(os.path.join(root_dir, "*")))
|
| 146 |
for sub_folder in tqdm(sub_folders, desc="checking scenes..."):
|
| 147 |
# img_dir = os.path.join(sub_folder, 'images_8')
|
| 148 |
img_dir = os.path.join(sub_folder, "images_4")
|