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

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
pull/8854/head
Johnny 11 months ago committed by GitHub
parent 7451ca1f54
commit 5893d4483e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
  1. 4
      ultralytics/data/base.py

@ -120,7 +120,9 @@ class BaseDataset(Dataset):
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]):

Loading…
Cancel
Save