Update fraction arg to employ a random selection (ultralytics#8234)

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
hmurari · Apr 17, 2024 · 22bfb48 · 22bfb48
1 parent 8dff0ca
commit 22bfb48
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/ultralytics/data/base.py b/ultralytics/data/base.py
@@ -120,7 +120,9 @@ def get_img_files(self, img_path):
         except Exception as e:
             raise FileNotFoundError(f"{self.prefix}Error loading data from {img_path}\n{HELP_URL}") from e
         if self.fraction < 1:
-            im_files = im_files[: round(len(im_files) * self.fraction)]
+            # im_files = im_files[: round(len(im_files) * self.fraction)]
+            num_elements_to_select = round(len(im_files) * self.fraction)
+            im_files = random.sample(im_files, num_elements_to_select)
         return im_files
 
     def update_labels(self, include_class: Optional[list]):