Clean-up

wpeebles · Feb 2, 2022 · 1f64782 · 1f64782
1 parent 3da3423
commit 1f64782
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/prepare_data.py b/prepare_data.py
@@ -251,7 +251,7 @@ def load_spair_data(path, size, out_path, category='cat', split='test'):
 
 
 def load_image_folder(path, pattern):
-    files = sorted(glob(f'{path}/{pattern}'))[420:1980]
+    files = sorted(glob(f'{path}/{pattern}'))
     bboxes = [None] * len(files)  # This means no bounding boxes are used
     return files, bboxes