Fixed bug that loaded data in subdirectories twice

NVlabs · Jul 14, 2024 · 7544636 · 7544636
1 parent e85dadb
commit 7544636
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/train2/utils_dope.py b/train2/utils_dope.py
@@ -181,11 +181,8 @@ def __init__(
 
         def load_data(path,extensions):
             imgs = loadimages(path,extensions = extensions)
-
-            # Check all the folders in path 
-            for name in os.listdir(str(path)):
-                imgs += loadimages(path +"/"+name,extensions = extensions)
             return imgs
+
         self.imgs = []
         for path_look in path_dataset:
             self.imgs += load_data(path_look,extensions = self.extensions)