diff --git a/dataloaders/dataset.py b/dataloaders/dataset.py index b5e01058..cad85eb6 100644 --- a/dataloaders/dataset.py +++ b/dataloaders/dataset.py @@ -125,8 +125,8 @@ def preprocess(self): file_path = os.path.join(self.root_dir, file) video_files = [name for name in os.listdir(file_path)] - train_and_valid, test = train_test_split(video_files, test_size=0.2, random_state=42) - train, val = train_test_split(train_and_valid, test_size=0.2, random_state=42) + #train_and_valid, test = train_test_split(video_files, test_size=0.2, random_state=42) + #train, val = train_test_split(train_and_valid, test_size=0.2, random_state=42) train_dir = os.path.join(self.output_dir, 'train', file) val_dir = os.path.join(self.output_dir, 'val', file) @@ -254,4 +254,4 @@ def crop(self, buffer, clip_len, crop_size): print(labels) if i == 1: - break \ No newline at end of file + break